diff --git a/tests/fakes/fake_trained_models_repository.py b/tests/fakes/fake_trained_models_repository.py
index aea7f53..58a978c 100644
--- a/tests/fakes/fake_trained_models_repository.py
+++ b/tests/fakes/fake_trained_models_repository.py
@@ -3,6 +3,7 @@
 from src.spira_training.shared.ports.trained_models_repository import (
     TrainedModelsRepository,
 )
+from tests.fakes.fake_model import FakeModel
 
 
 class FakeTrainedModelsRepository(TrainedModelsRepository):
@@ -18,4 +19,4 @@ async def save_model(self, model: TrainedModel, path: Path) -> None:
 
 
 def make_trained_model():
-    return TrainedModel()
+    return FakeModel()