diff --git a/MLStructFP_benchmarks/ml/model/core/_model.py b/MLStructFP_benchmarks/ml/model/core/_model.py
index 81f582b..435beb8 100644
--- a/MLStructFP_benchmarks/ml/model/core/_model.py
+++ b/MLStructFP_benchmarks/ml/model/core/_model.py
@@ -1280,7 +1280,7 @@ def _custom_fit_tensor(
                 the generator as they can't be passed easily to children processes.
         :param kwargs: Optional keyword arguments
 
-        :returns:
+        :return:
             A `History` object. Its `History.history` attribute is
             a record of training loss values and metrics values
             at successive epochs, as well as validation loss values