diff --git a/src_web/comfyui/rgthree.ts b/src_web/comfyui/rgthree.ts
index f756935..9034f23 100644
--- a/src_web/comfyui/rgthree.ts
+++ b/src_web/comfyui/rgthree.ts
@@ -655,7 +655,7 @@ class Rgthree extends EventTarget {
           }, 5000);
         }
       }, 100);
-      loadGraphData && loadGraphData.call(app, ...arguments);
+      return loadGraphData && loadGraphData.call(app, ...arguments);
     };
   }
 
diff --git a/web/comfyui/rgthree.js b/web/comfyui/rgthree.js
index 8be1d78..64a093d 100644
--- a/web/comfyui/rgthree.js
+++ b/web/comfyui/rgthree.js
@@ -473,7 +473,7 @@ class Rgthree extends EventTarget {
                     }, 5000);
                 }
             }, 100);
-            loadGraphData && loadGraphData.call(app, ...arguments);
+            return loadGraphData && loadGraphData.call(app, ...arguments);
         };
     }
     getNodeFromInitialGraphToPromptSerializedWorkflowBecauseComfyUIBrokeStuff(node) {