diff --git a/package-lock.json b/package-lock.json
index 3b0b6dfae..b9d15eb07 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -9,7 +9,7 @@
       "version": "1.4.6",
       "dependencies": {
         "@atlaskit/pragmatic-drag-and-drop": "^1.3.1",
-        "@comfyorg/comfyui-electron-types": "^0.3.1",
+        "@comfyorg/comfyui-electron-types": "^0.3.6",
         "@comfyorg/litegraph": "^0.8.31",
         "@primevue/themes": "^4.0.5",
         "@vueuse/core": "^11.0.0",
@@ -1917,9 +1917,10 @@
       "dev": true
     },
     "node_modules/@comfyorg/comfyui-electron-types": {
-      "version": "0.3.2",
-      "resolved": "https://registry.npmjs.org/@comfyorg/comfyui-electron-types/-/comfyui-electron-types-0.3.2.tgz",
-      "integrity": "sha512-3vpAAZq4rNd6oUFvX5s1kor2U7xN3Br8gftdO40IvhLLikttK2cZhPra6YlnzuYIU1gK+PBGEJClvjFnQyIYlA=="
+      "version": "0.3.6",
+      "resolved": "https://registry.npmjs.org/@comfyorg/comfyui-electron-types/-/comfyui-electron-types-0.3.6.tgz",
+      "integrity": "sha512-wgMgESnCcRzvVkk8CwWiTAUJxC4LBvw5uTENxzaWkEL0qrnmiGrVLore00yX3cYz04hJaTA6PqasLqgVLDOenw==",
+      "license": "GPL-3.0-only"
     },
     "node_modules/@comfyorg/litegraph": {
       "version": "0.8.31",
diff --git a/package.json b/package.json
index 7f6d2e712..6e3c7ea37 100644
--- a/package.json
+++ b/package.json
@@ -72,7 +72,7 @@
   },
   "dependencies": {
     "@atlaskit/pragmatic-drag-and-drop": "^1.3.1",
-    "@comfyorg/comfyui-electron-types": "^0.3.1",
+    "@comfyorg/comfyui-electron-types": "^0.3.6",
     "@comfyorg/litegraph": "^0.8.31",
     "@primevue/themes": "^4.0.5",
     "@vueuse/core": "^11.0.0",
diff --git a/src/stores/electronDownloadStore.ts b/src/stores/electronDownloadStore.ts
index 93d3aa6f4..c5757a88b 100644
--- a/src/stores/electronDownloadStore.ts
+++ b/src/stores/electronDownloadStore.ts
@@ -1,7 +1,10 @@
 import { ref } from 'vue'
 import { defineStore } from 'pinia'
 import { isElectron, electronAPI } from '@/utils/envUtil'
-import { DownloadState, DownloadStatus } from '@comfyorg/comfyui-electron-types'
+import type {
+  DownloadState,
+  DownloadStatus
+} from '@comfyorg/comfyui-electron-types'
 
 export interface ElectronDownload
   extends Pick<DownloadState, 'url' | 'filename'> {
diff --git a/vite.config.mts b/vite.config.mts
index a64062c7d..7d454479b 100644
--- a/vite.config.mts
+++ b/vite.config.mts
@@ -150,10 +150,7 @@ export default defineConfig({
     rollupOptions: {
       // Disabling tree-shaking
       // Prevent vite remove unused exports
-      treeshake: false,
-      external: [
-        'electron'
-      ]
+      treeshake: false
     }
   },