diff --git a/src/types/index.ts b/src/types/index.ts
index b8d6a81ae..8b6242087 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -40,6 +40,7 @@ export type {
   parseNodeExecutionId,
   createNodeExecutionId
 } from './nodeIdentification'
+export type { DOMWidget, DOMWidgetOptions } from '@/scripts/domWidget'
 export type {
   EmbeddingsResponse,
   ExtensionsResponse,
diff --git a/src/types/litegraph-augmentation.d.ts b/src/types/litegraph-augmentation.d.ts
index 2e04dc501..1151cc7a8 100644
--- a/src/types/litegraph-augmentation.d.ts
+++ b/src/types/litegraph-augmentation.d.ts
@@ -1,5 +1,10 @@
 import '@/lib/litegraph/src/litegraph'
 import type { LLink, Size } from '@/lib/litegraph/src/litegraph'
+import type {
+  ExecutableLGraphNode,
+  ExecutionId
+} from '@/lib/litegraph/src/litegraph'
+import type { IBaseWidget } from '@/lib/litegraph/src/litegraph/dist/types/widgets'
 import type { ComfyNodeDef as ComfyNodeDefV2 } from '@/schemas/nodeDef/nodeDefSchemaV2'
 import type { ComfyNodeDef as ComfyNodeDefV1 } from '@/schemas/nodeDefSchema'
 import type { DOMWidget, DOMWidgetOptions } from '@/scripts/domWidget'
@@ -68,12 +73,6 @@ declare module '@/lib/litegraph/src/interfaces' {
  *  ComfyUI extensions of litegraph
  */
 declare module '@/lib/litegraph/src/litegraph' {
-  import type {
-    ExecutableLGraphNode,
-    ExecutionId
-  } from '@/lib/litegraph/src/litegraph'
-  import type { IBaseWidget } from '@/lib/litegraph/src/types/widgets'
-
   interface LGraphNodeConstructor<T extends LGraphNode = LGraphNode> {
     type?: string
     comfyClass: string