diff --git a/pyproject_comfyui.toml b/pyproject_comfyui.toml
index d3b3c18..d85bf99 100644
--- a/pyproject_comfyui.toml
+++ b/pyproject_comfyui.toml
@@ -20,5 +20,6 @@ dependencies = [
     "imageio",
     "loguru",
     "pydantic",
-    "pydantic_settings"
+    "pydantic_settings",
+    "backoff"
 ]
\ No newline at end of file
diff --git a/src/cluster/comfyui_api.py b/src/cluster/comfyui_api.py
index f68035a..bcca8d4 100644
--- a/src/cluster/comfyui_api.py
+++ b/src/cluster/comfyui_api.py
@@ -1,7 +1,5 @@
 import subprocess
-
 import modal
-from cluster.comfyui_v1 import comfyui_image
 
 image = (  # build up a Modal Image to run ComfyUI, step by step
     modal.Image.debian_slim(  # start from basic Linux with Python