diff --git a/main.py b/main.py
index d174744507e3fcd34512e5a09e899e450c4c32ff..afaf5ad93073106b73513435ae1adea5311323d3 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,4 @@
-from src.tree import TreeWalker
+from skill_tree.tree import TreeWalker
 
 if __name__ == '__main__':
     walker = TreeWalker("data", "java", "Java")