diff --git a/test/e2e/helpers/console-helpers.ts b/test/e2e/helpers/console-helpers.ts
index c5f5191106b..63434fb0b4f 100644
--- a/test/e2e/helpers/console-helpers.ts
+++ b/test/e2e/helpers/console-helpers.ts
@@ -135,7 +135,7 @@ export async function typeIntoConsole(frontend: puppeteer.Page, message: string)
   // The autocomplete element doesn't exist until the first autocomplete suggestion
   // is actaully given.
   const autocomplete = autocompleteHandle ? autocompleteHandle.asElement()! : null;
-  await frontend.waitFor(
+  await frontend.waitForFunction(
       (msg, ln, ac) => ln.textContent === msg && (!ac || ac.textContent === ''), undefined, message, line,
       autocomplete);
 
diff --git a/test/e2e/sources/script-in-multiple-workers_test.ts b/test/e2e/sources/script-in-multiple-workers_test.ts
index f17714043e4..9550188a5ad 100644
--- a/test/e2e/sources/script-in-multiple-workers_test.ts
+++ b/test/e2e/sources/script-in-multiple-workers_test.ts
@@ -117,7 +117,7 @@ describe('Multi-Workers', async function() {
           return;
         }
         await bpCheckbox.evaluate(n => (n as HTMLElement).click());
-        await frontend.waitFor('.cm-breakpoint-disabled');
+        await frontend.waitForSelector('.cm-breakpoint-disabled');
         // Add another breakpoint
         await addBreakpointForLine(frontend, 12);