diff --git a/tsp/tree_model.py b/tree_model.py
similarity index 100%
rename from tsp/tree_model.py
rename to tree_model.py
diff --git a/tsp-cli-client b/tsp-cli-client
index 5664b66..17e6ede 100755
--- a/tsp-cli-client
+++ b/tsp-cli-client
@@ -34,7 +34,7 @@ from datetime import timedelta
 from decimal import Decimal
 from os.path import os
 
-from tsp.tree_model import TreeModel
+from tree_model import TreeModel
 from tsp.tsp_client import TspClient
 from tsp.xy_model import XYModel