diff --git a/comfyui_to_python.py b/comfyui_to_python.py
index ec9566e..d654069 100644
--- a/comfyui_to_python.py
+++ b/comfyui_to_python.py
@@ -11,7 +11,8 @@ from argparse import ArgumentParser
 
 import black
 
-
+# Add current directory to Python path
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 from comfyui_to_python_utils import (
     import_custom_nodes,
     find_path,
diff --git a/comfyui_to_python_utils.py b/comfyui_to_python_utils.py
index 12b3be2..0aefa4c 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
@@ -23,7 +24,7 @@ def import_custom_nodes() -> None:
     execution.PromptQueue(server_instance)
 
     # Initializing custom nodes
-    init_extra_nodes()
+    asyncio.run(init_extra_nodes())
 
 
 def find_path(name: str, path: str = None) -> str: