diff --git a/comfyui_to_python_utils.py b/comfyui_to_python_utils.py
index 12b3be2..cff4393 100644
--- a/comfyui_to_python_utils.py
+++ b/comfyui_to_python_utils.py
@@ -12,6 +12,7 @@ def import_custom_nodes() -> None:
     import asyncio
     import execution
     from nodes import init_extra_nodes
+    sys.path.insert(0, find_path("ComfyUI"))
     import server
 
     # Creating a new event loop and setting it as the default loop