diff --git a/baybe/surrogates/utils.py b/baybe/surrogates/utils.py
index 271d0eb900..8d28f2357a 100644
--- a/baybe/surrogates/utils.py
+++ b/baybe/surrogates/utils.py
@@ -99,10 +99,10 @@ def _fit_new(
         if train_y.numel() == 1 or train_y.std() < std_threshold:
             self._model = MeanPredictionSurrogate()
             self._model._fit(searchspace, train_x, train_y)
-            return
 
         # Regular operation
-        _fit_original(self, searchspace, train_x, train_y)
+        else:
+            _fit_original(self, searchspace, train_x, train_y)
 
     # Replace the methods
     cls._posterior = _posterior_new