diff --git a/stepcovnet/executor.py b/stepcovnet/executor.py
index 57183c2..1095761 100644
--- a/stepcovnet/executor.py
+++ b/stepcovnet/executor.py
@@ -117,7 +117,7 @@ def execute(self, input_data: inputs.TrainingInput) -> model.StepCOVNetModel:
                 input_data,
                 saved_original_weights=weights,
                 epochs=epochs_final,
-                callbacks=self.get_retraining_callbacks(hyperparameters),
+                callback_list=self.get_retraining_callbacks(hyperparameters),
             )
             self.save(
                 input_data.config, training_history=retraining_history, retrained=True