diff --git a/browser/components/urlbar/tests/UrlbarTestUtils.sys.mjs b/browser/components/urlbar/tests/UrlbarTestUtils.sys.mjs
index 325491742183..88bae0fedbe1 100644
--- a/browser/components/urlbar/tests/UrlbarTestUtils.sys.mjs
+++ b/browser/components/urlbar/tests/UrlbarTestUtils.sys.mjs
@@ -1151,7 +1151,7 @@ export var UrlbarTestUtils = {
         urlbarValue,
         "Urlbar value hasn't changed."
       );
-      this.assertSearchMode(window, null);
+      await this.assertSearchMode(window, null);
     } else if (clickClose) {
       // We need to hover the indicator to make the close button clickable in the
       // test.
diff --git a/browser/components/urlbar/tests/browser/browser_searchFunction.js b/browser/components/urlbar/tests/browser/browser_searchFunction.js
index a4c0da20fe92..ab39a04423f5 100644
--- a/browser/components/urlbar/tests/browser/browser_searchFunction.js
+++ b/browser/components/urlbar/tests/browser/browser_searchFunction.js
@@ -49,7 +49,7 @@ add_task(async function searchEngineAlias() {
     gURLBar.search("@example")
   );
   ok(gURLBar.hasAttribute("focused"), "url bar is focused");
-  UrlbarTestUtils.assertSearchMode(window, null);
+  await UrlbarTestUtils.assertSearchMode(window, null);
   await assertUrlbarValue("@example");
 
   assertOneOffButtonsVisible(false);