diff --git a/model_manager/utils/repository/training_repository.py b/model_manager/utils/repository/training_repository.py
index 228f7a7..6a28b3e 100644
--- a/model_manager/utils/repository/training_repository.py
+++ b/model_manager/utils/repository/training_repository.py
@@ -18,6 +18,8 @@ from model_manager.sientia.metrics import mae, mse, r2
 from model_manager.sientia.models import (
     DataPreprocessor,
     LinearRegressionModel,
+)
+from model_manager.sientia.models import (
     _frontend_date_format_to_strftime as _frontend_format_to_strftime,
 )
 from model_manager.sientia.utils import split_train_test