diff --git a/README.md b/README.md
index 16502ef..372eaea 100644
--- a/README.md
+++ b/README.md
@@ -90,13 +90,6 @@ def prompt_worker(q, server):
         item, item_id = q.get()
         # Add next line
         global_bpy.reset_bpy()
-
-if __name__ == "__main__":
-    # Add next two lines 
-    import sys
-    sys.path.append(os.path.join(os.path.dirname(__file__), "custom_nodes", "avatar-graph-comfyui"))
-
-
 ```
 
 5. Restart comfyui
diff --git a/__init__.py b/__init__.py
index e0dd9ae..f7ece75 100644
--- a/__init__.py
+++ b/__init__.py
@@ -4,11 +4,13 @@
 @nickname: Avatar Graph
 @description: Include nodes for sam + bpy operation, that allows workflow creations for generative 2d character rig.
 """
-import routes
-
-import inspect
 import os
 import sys
+sys.path.append(os.path.join(os.path.dirname(__file__)))
+
+import routes
+import inspect
+import sys
 import importlib
 import subprocess
 from folder_paths import add_model_folder_path