diff --git a/browser_tests/tests/dynamicWidgetsSubgraph.spec.ts b/browser_tests/tests/dynamicWidgetsSubgraph.spec.ts
index 416999e9b..62b6a4d64 100644
--- a/browser_tests/tests/dynamicWidgetsSubgraph.spec.ts
+++ b/browser_tests/tests/dynamicWidgetsSubgraph.spec.ts
@@ -310,7 +310,9 @@ test.describe('Dynamic Combo Widgets in Subgraphs', () => {
 
     await comfyPage.page.keyboard.press('Escape')
     await comfyPage.nextFrame()
-    expect(await comfyPage.isInSubgraph()).toBe(false)
+    await expect
+      .poll(() => comfyPage.isInSubgraph(), { timeout: 5000 })
+      .toBe(false)
 
     const outerSubgraph = await getSubgraphNode(comfyPage)
     expect(await outerSubgraph.getWidgets()).toEqual([