diff --git a/browser_tests/tests/interaction.spec.ts b/browser_tests/tests/interaction.spec.ts
index eec7d64dc..14c082707 100644
--- a/browser_tests/tests/interaction.spec.ts
+++ b/browser_tests/tests/interaction.spec.ts
@@ -702,6 +702,24 @@ test.describe('Load workflow', { tag: '@screenshot' }, () => {
     await expect(comfyPage.canvas).toHaveScreenshot(
       'single_ksampler_modified.png'
     )
+    // Wait for V2 persistence to save the modified workflow (debounced at 512ms)
+    // Check that localStorage has a draft key with the collapsed node state
+    await comfyPage.page.waitForFunction(
+      () => {
+        for (let i = 0; i < window.localStorage.length; i++) {
+          const key = window.localStorage.key(i)
+          if (key?.startsWith('Comfy.Draft:')) {
+            const value = window.localStorage.getItem(key)
+            // Check that the draft contains collapsed node flags
+            if (value && value.includes('"flags":')) {
+              return true
+            }
+          }
+        }
+        return false
+      },
+      { timeout: 2000 }
+    )
     await comfyPage.setup({ clearStorage: false })
     await expect(comfyPage.canvas).toHaveScreenshot(
       'single_ksampler_modified.png'