diff --git a/comfyui_custom_scripts/extensions/webuiNodes.js b/comfyui_custom_scripts/extensions/webuiNodes.js
index dc4fa87..af97722 100644
--- a/comfyui_custom_scripts/extensions/webuiNodes.js
+++ b/comfyui_custom_scripts/extensions/webuiNodes.js
@@ -68,15 +68,6 @@ app.registerExtension({
             }
         }
     },
-    async setup() {
-        try {
-            await iframeRegisteredEvent;
-        } catch {
-            return;
-        }
-
-        app.loadGraphData();
-    },
     async nodeCreated(node) {
         let iframeInfo = null;
 
@@ -97,6 +88,9 @@ app.registerExtension({
         // 240 and 40 are empirical values that seem to work
         node.size = [240, 40 + distanceBetweenIoSlots * maxIoLength];
     },
+    async setup() {
+        app.loadGraphData();
+    },
 });
 
 const webuiIoNodeNames = [