diff --git a/tests/test_git.py b/tests/test_git.py
index 43134e7..341c230 100644
--- a/tests/test_git.py
+++ b/tests/test_git.py
@@ -3,11 +3,9 @@ import sys
 import os
 sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))))
 try:
-    from custom_nodes.EternalKernelLiteGraphNodes.git import CreateRepoNode
-except:
-    from ..git import CreateRepoNode
-
-
+    from custom_nodes.git import CreateRepoNode
+except ImportError:
+    from custom_nodes.git import CreateRepoNode
 
 class TestCreateRepoNode(unittest.TestCase):
     def test_create_repo(self):
@@ -15,6 +13,5 @@ class TestCreateRepoNode(unittest.TestCase):
         node.create_repo(subdirectory="my_repo")
         # Add assertions or additional test logic here
 
-
 if __name__ == "__main__":
     unittest.main()