diff --git a/src/helios/model/model.py b/src/helios/model/model.py
index 52b33e3..ebac310 100644
--- a/src/helios/model/model.py
+++ b/src/helios/model/model.py
@@ -211,7 +211,6 @@ def on_training_batch_start(self, state: TrainingState) -> None:
         """
         self._loss_items.clear()
 
-    @abc.abstractmethod
     def train_step(self, batch: typing.Any, state: TrainingState) -> None:
         """
         Perform a single training step.