diff --git a/src/ai/__init__.py b/src/ai/__init__.py
index 19c9b57..7bc8b97 100644
--- a/src/ai/__init__.py
+++ b/src/ai/__init__.py
@@ -1,5 +1,5 @@
-from .config import get_config
-from .io import read_genome
-from .training import train
-
-__all__ = ["train", "read_genome", "get_config"]
+# from .config import get_config
+# from .io import read_genome
+# from .training import train
+#
+# __all__ = ["train", "read_genome", "get_config"]
diff --git a/src/ai/config.py b/src/ai/config.py
index c111479..f761d5a 100644
--- a/src/ai/config.py
+++ b/src/ai/config.py
@@ -1,5 +1,5 @@
 import neat
-from path import BASE_PATH
+from utils import BASE_PATH
 
 
 def get_config() -> neat.Config:
diff --git a/src/ai/io.py b/src/ai/io.py
index d5c7a45..6bc7e65 100644
--- a/src/ai/io.py
+++ b/src/ai/io.py
@@ -2,7 +2,7 @@ import pickle
 from pathlib import Path
 
 import neat
-from path import BASE_PATH
+from utils import BASE_PATH
 
 
 def read_genome() -> neat.DefaultGenome:
diff --git a/src/ai/training.py b/src/ai/training.py
index 83384ad..b5078cc 100644
--- a/src/ai/training.py
+++ b/src/ai/training.py
@@ -1,6 +1,6 @@
 import neat
 from loguru import logger
-from path import BASE_PATH
+from utils import BASE_PATH
 
 from .config import get_config
 from .evaluation import eval_genomes