diff --git a/browser_tests/tests/interaction.spec.ts b/browser_tests/tests/interaction.spec.ts
index 14c082707..6380db714 100644
--- a/browser_tests/tests/interaction.spec.ts
+++ b/browser_tests/tests/interaction.spec.ts
@@ -708,7 +708,7 @@ test.describe('Load workflow', { tag: '@screenshot' }, () => {
       () => {
         for (let i = 0; i < window.localStorage.length; i++) {
           const key = window.localStorage.key(i)
-          if (key?.startsWith('Comfy.Draft:')) {
+          if (key?.startsWith('Comfy.Workflow.Draft.v2:')) {
             const value = window.localStorage.getItem(key)
             // Check that the draft contains collapsed node flags
             if (value && value.includes('"flags":')) {