diff --git a/src/__init__.py b/src/__init__.py
index fe41f9d..e1e8f64 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -4,5 +4,5 @@ if 'AI' in os.getcwd():
     from src.TSP_solver import *
     from src.io_tsp import *
 else:
-    from AI2019.code.TSP_solver import *
-    from AI2019.code.io_tsp import *
\ No newline at end of file
+    from AI2019.src.TSP_solver import *
+    from AI2019.src.io_tsp import *
\ No newline at end of file