diff --git a/theseus/base/callbacks/__init__.py b/theseus/base/callbacks/__init__.py
index 3d1da88..1de832e 100644
--- a/theseus/base/callbacks/__init__.py
+++ b/theseus/base/callbacks/__init__.py
@@ -5,7 +5,7 @@
     RichModelSummary,
 )
 
-from theseus.ml.callbacks.base_callbacks import Callbacks, CallbacksList
+# from theseus.ml.callbacks.base_callbacks import Callbacks, CallbacksList
 from theseus.registry import Registry
 
 from .checkpoint_callback import TorchCheckpointCallback
diff --git a/theseus/ml/trainer/ml_trainer.py b/theseus/ml/trainer/ml_trainer.py
index 2c64b93..00bbe0a 100644
--- a/theseus/ml/trainer/ml_trainer.py
+++ b/theseus/ml/trainer/ml_trainer.py
@@ -1,5 +1,5 @@
-from theseus.base.callbacks import CallbacksList
 from theseus.base.utilities.loggers.observer import LoggerObserver
+from theseus.ml.callbacks import CallbacksList
 
 LOGGER = LoggerObserver.getLogger("main")