diff --git a/src/extensions/core/index.ts b/src/extensions/core/index.ts
index 00c925935..5354ef4e9 100644
--- a/src/extensions/core/index.ts
+++ b/src/extensions/core/index.ts
@@ -11,7 +11,6 @@ import './maskeditor'
 import './nodeTemplates'
 import './noteNode'
 import './previewAny'
-import './proxyWidget'
 import './rerouteNode'
 import './saveImageExtraOutput'
 import './saveMesh'
diff --git a/src/extensions/core/proxyWidget.ts b/src/scripts/proxyWidget.ts
similarity index 99%
rename from src/extensions/core/proxyWidget.ts
rename to src/scripts/proxyWidget.ts
index 3b04a3a2c..c5aad76a6 100644
--- a/src/extensions/core/proxyWidget.ts
+++ b/src/scripts/proxyWidget.ts
@@ -7,10 +7,10 @@ import { DOMWidgetImpl } from '@/scripts/domWidget'
 import { useDomWidgetStore } from '@/stores/domWidgetStore'
 import { useCanvasStore } from '@/stores/graphStore'
 
-const canvasStore = useCanvasStore()
 const originalConfigureAfterSlots =
   SubgraphNode.prototype._internalConfigureAfterSlots
 SubgraphNode.prototype._internalConfigureAfterSlots = function () {
+  const canvasStore = useCanvasStore()
   const subgraphNode = this
   //Must give value to proxyWidgets prior to injecting or it won't serialize
   subgraphNode.properties.proxyWidgets ??= '[]'
diff --git a/src/scripts/widgets.ts b/src/scripts/widgets.ts
index 357dc2bf1..50944381a 100644
--- a/src/scripts/widgets.ts
+++ b/src/scripts/widgets.ts
@@ -29,6 +29,7 @@ import { useSettingStore } from '@/stores/settingStore'
 import type { ComfyApp } from './app'
 import './domWidget'
 import './errorNodeWidgets'
+import './proxyWidget'
 
 export type ComfyWidgetConstructorV2 = (
   node: LGraphNode,