diff --git a/tests/test_git.py b/tests/test_git.py
index aa4c355..7de04f2 100644
--- a/tests/test_git.py
+++ b/tests/test_git.py
@@ -1,5 +1,9 @@
 import unittest
-from custom_nodes.EternalKernelLiteGraphNodes.git import CreateRepoNode
+try:
+    from custom_nodes.EternalKernelLiteGraphNodes.git import CreateRepoNode
+except:
+    from ..git import CreateRepoNode
+
 
 
 class TestCreateRepoNode(unittest.TestCase):