diff --git a/package.json b/package.json
index 429d5b3f..1474a20c 100644
--- a/package.json
+++ b/package.json
@@ -3,7 +3,7 @@
   "productName": "ComfyUI",
   "repository": "github:comfy-org/electron",
   "copyright": "Copyright © 2024 Comfy Org",
-  "version": "0.3.5",
+  "version": "0.3.6",
   "homepage": "https://comfy.org",
   "description": "The best modular GUI to run AI diffusion models.",
   "main": ".vite/build/main.js",
diff --git a/src/main_types.ts b/src/main_types.ts
index b03d0dc5..643cd075 100644
--- a/src/main_types.ts
+++ b/src/main_types.ts
@@ -1,3 +1,3 @@
 export * from './constants';
-export type { DownloadManager, Download } from './models/DownloadManager';
+export type { DownloadManager, Download, DownloadState } from './models/DownloadManager';
 export type { ElectronAPI } from './preload';