diff --git a/src/comfy_pack/run.py b/src/comfy_pack/run.py
index 02f875b..3c0d1dc 100644
--- a/src/comfy_pack/run.py
+++ b/src/comfy_pack/run.py
@@ -74,6 +74,9 @@ class ComfyUIServer:
         self.temp_dir = run_dir / "temp"
         self.output_dir = run_dir / "output"
 
+        self.temp_dir.mkdir(parents=True, exist_ok=True)
+        self.output_dir.mkdir(parents=True, exist_ok=True)
+
         if port is None:
             if _is_port_in_use(8188):
                 self.port = port if port else random.randint(58000, 58999)