Index: ResultsManager.js
===================================================================
--- ResultsManager.js	(revision 5781)
+++ ResultsManager.js	(working copy)
@@ -68,7 +68,7 @@
             this.fields[formIndex][fieldIndex] = new FieldResult(formIndex, fieldIndex);
         }
         this.fields[formIndex][fieldIndex].addResults(resultsWrapper.results);
-        this.extensionManager.warningDialog.finishedTest();
+        this.extensionManager.finishedTest();
         if (this.attacks.length === 0 && this.sourceListeners.length === 0) {
             dump('\nall results logged now.')
             this.allResultsLogged = true;
Index: TestManager.js
===================================================================
--- TestManager.js	(revision 5781)
+++ TestManager.js	(working copy)
@@ -116,7 +116,7 @@
             
         }
         
-        this.controller.warningDialog.finishedTest();
+        this.controller.finishedTest();
         
         this.resultsStillExpected--;
     }
Index: xssme_sidebar.js
===================================================================
--- xssme_sidebar.js	(revision 5782)
+++ xssme_sidebar.js	(working copy)
@@ -615,6 +615,16 @@
         return rv;
         
     }
+    ,
+    /**
+     * Called by the testmanager and resultsmanager to report that a test has
+     * been completed so that the popup's UI can be updated
+     */
+    finishedTest: function() {
+        if (this.warningDialog) {
+            this.warningDialog.finishedTest();
+        }
+    }
 }