diff --git a/src/TSP_solver.py b/src/TSP_solver.py
index d305a73..c08820b 100644
--- a/src/TSP_solver.py
+++ b/src/TSP_solver.py
@@ -1,6 +1,10 @@
 import numpy as np
 from matplotlib import pyplot as plt
 from numpy.core._multiarray_umath import ndarray
+if 'AI' in os.getcwd():
+    from src.utils import *
+else:
+    from AI2019.src.utils import *
 
 
 
diff --git a/src/__init__.py b/src/__init__.py
index 175bd3e..3bc0b72 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -5,6 +5,6 @@ if 'AI' in os.getcwd():
     from src.TSP_solver import *
     from src.io_tsp import *
 else:
-    from AI2019src.utils import *
+    from AI2019.src.utils import *
     from AI2019.src.TSP_solver import *
     from AI2019.src.io_tsp import *