1 files changed, 7 insertions, 5 deletions
diff --git a/uriloader/exthandler/tests/mochitest/browser_txt_download_save_as.js b/uriloader/exthandler/tests/mochitest/browser_txt_download_save_as.js
index 7155c35fd9..9d0f8db41c 100644
--- a/uriloader/exthandler/tests/mochitest/browser_txt_download_save_as.js
+++ b/uriloader/exthandler/tests/mochitest/browser_txt_download_save_as.js
@@ -25,7 +25,7 @@ const {
 } = Ci.nsIHandlerInfo;
 const testDir = createTemporarySaveDirectory();
 const MockFilePicker = SpecialPowers.MockFilePicker;
-MockFilePicker.init(window);
+MockFilePicker.init(window.browsingContext);
 
 Services.scriptloader.loadSubScript(
   "chrome://mochitests/content/browser/toolkit/content/tests/browser/common/mockTransfer.js",
@@ -64,12 +64,14 @@ async function setupFilePicker() {
     MockFilePicker.displayDirectory = testDir;
     MockFilePicker.showCallback = fp => {
       ok(true, "filepicker should be visible");
-      ok(
-        fp.defaultExtension === "txt",
+      Assert.strictEqual(
+        fp.defaultExtension,
+        "txt",
         "Default extension in filepicker should be txt"
       );
-      ok(
-        fp.defaultString === "file_txt_attachment_test.txt",
+      Assert.strictEqual(
+        fp.defaultString,
+        "file_txt_attachment_test.txt",
         "Default string name in filepicker should have the correct file name"
       );
       const destFile = testDir.clone();