diff --git a/package-lock.json b/package-lock.json
index 6066f0fed..063f27cc0 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -11,7 +11,7 @@
       "dependencies": {
         "@atlaskit/pragmatic-drag-and-drop": "^1.3.1",
         "@comfyorg/comfyui-electron-types": "^0.4.20",
-        "@comfyorg/litegraph": "^0.8.94",
+        "@comfyorg/litegraph": "^0.8.96",
         "@primevue/forms": "^4.2.5",
         "@primevue/themes": "^4.2.5",
         "@sentry/vue": "^8.48.0",
@@ -1944,9 +1944,9 @@
       "license": "GPL-3.0-only"
     },
     "node_modules/@comfyorg/litegraph": {
-      "version": "0.8.94",
-      "resolved": "https://registry.npmjs.org/@comfyorg/litegraph/-/litegraph-0.8.94.tgz",
-      "integrity": "sha512-/eKe4sufoZ2Non3tfdyA89DgxpMJ55U+uHP82Hj4AoGH1O3ZhGV9KG6B0XIdiAfqSzqkMIMVazYCYwlWSHA/cg==",
+      "version": "0.8.96",
+      "resolved": "https://registry.npmjs.org/@comfyorg/litegraph/-/litegraph-0.8.96.tgz",
+      "integrity": "sha512-t9vlj8zVXwVhgjdZlOioNXY+AJADknEYhsWwbMmmRU4oQVskhWzrK3G/OqNOMilDRla+/oinqoFmVg4ujE9MVA==",
       "license": "MIT"
     },
     "node_modules/@cspotcode/source-map-support": {
diff --git a/package.json b/package.json
index 9a0a4b406..89d9e9231 100644
--- a/package.json
+++ b/package.json
@@ -84,7 +84,7 @@
   "dependencies": {
     "@atlaskit/pragmatic-drag-and-drop": "^1.3.1",
     "@comfyorg/comfyui-electron-types": "^0.4.20",
-    "@comfyorg/litegraph": "^0.8.94",
+    "@comfyorg/litegraph": "^0.8.96",
     "@primevue/forms": "^4.2.5",
     "@primevue/themes": "^4.2.5",
     "@sentry/vue": "^8.48.0",
diff --git a/src/extensions/core/widgetInputs.ts b/src/extensions/core/widgetInputs.ts
index 9aec039fc..4827e8294 100644
--- a/src/extensions/core/widgetInputs.ts
+++ b/src/extensions/core/widgetInputs.ts
@@ -515,6 +515,7 @@ export function convertToInput(
   const [oldWidth, oldHeight] = node.size
   const inputIsOptional = !!widget.options?.inputIsOptional
   const input = node.addInput(widget.name, type, {
+    // @ts-expect-error [GET_CONFIG] is not a valid property of IWidget
     widget: { name: widget.name, [GET_CONFIG]: () => config },
     ...(inputIsOptional ? { shape: LiteGraph.SlotShape.HollowCircle } : {})
   })
diff --git a/src/services/litegraphService.ts b/src/services/litegraphService.ts
index e571cbd8e..8b28b2452 100644
--- a/src/services/litegraphService.ts
+++ b/src/services/litegraphService.ts
@@ -4,7 +4,8 @@ import {
   type INodeInputSlot,
   LGraphEventMode,
   LGraphNode,
-  LiteGraph
+  LiteGraph,
+  RenderShape
 } from '@comfyorg/litegraph'
 import { Vector2 } from '@comfyorg/litegraph'
 import { IBaseWidget, IWidget } from '@comfyorg/litegraph/dist/types/widgets'
@@ -89,7 +90,7 @@ export const useLitegraphService = () => {
             // Node connection inputs
             const shapeOptions = inputIsRequired
               ? {}
-              : { shape: LiteGraph.SlotShape.HollowCircle }
+              : { shape: RenderShape.HollowCircle }
             const inputOptions = {
               ...shapeOptions,
               localized_name: st(nameKey, inputName)