diff --git a/image.py b/image.py
index 0b410c3..88b9d33 100644
--- a/image.py
+++ b/image.py
@@ -1,5 +1,5 @@
 import os
-
+import comfy
 try:
     from line_profiler_pycharm import profile
 except ImportError as e: