diff --git a/nodes.py b/nodes.py
index d24df20..453f6e6 100644
--- a/nodes.py
+++ b/nodes.py
@@ -15,6 +15,9 @@ from PIL.PngImagePlugin import PngInfo
 import numpy as np
 import safetensors.torch
 
+sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
+
+
 import comfy.diffusers_load
 import comfy.samplers
 import comfy.sample