1 files changed, 3 insertions, 5 deletions
diff --git a/devtools/client/debugger/test/mochitest/browser_dbg-breakpoints-popup.js b/devtools/client/debugger/test/mochitest/browser_dbg-breakpoints-popup.js
index eba02919d3..8a7ac18df4 100644
--- a/devtools/client/debugger/test/mochitest/browser_dbg-breakpoints-popup.js
+++ b/devtools/client/debugger/test/mochitest/browser_dbg-breakpoints-popup.js
@@ -70,8 +70,7 @@ add_task(async function testPausedByBreakpoint() {
   info("Open the popup in order to be able to set a breakpoint");
   const firstPopupBrowsingContext = await openPopup(POPUP_URL);
 
-  await waitForSource(dbg, POPUP_URL);
-  const source = findSource(dbg, POPUP_URL);
+  let source = await waitForSource(dbg, POPUP_URL);
 
   await selectSource(dbg, source);
   await addBreakpoint(dbg, source, 4);
@@ -88,7 +87,7 @@ add_task(async function testPausedByBreakpoint() {
     "The popup is really paused"
   );
 
-  await waitForSource(dbg, POPUP_URL);
+  source = await waitForSource(dbg, POPUP_URL);
   assertPausedAtSourceAndLine(dbg, source.id, 4);
 
   await resume(dbg);
@@ -131,8 +130,7 @@ add_task(async function testPausedInTwoPopups() {
   const browser = gBrowser.selectedBrowser;
   const popupBrowsingContext = await openPopup(POPUP_URL);
 
-  await waitForSource(dbg, POPUP_URL);
-  const source = findSource(dbg, POPUP_URL);
+  const source = await waitForSource(dbg, POPUP_URL);
 
   await selectSource(dbg, source);
   await addBreakpoint(dbg, source, 4);