diff --git a/nnunetv2/training/nnUNetTrainer/variants/benchmarking/nnUNetTrainerBenchmark_5epochs.py b/nnunetv2/training/nnUNetTrainer/variants/benchmarking/nnUNetTrainerBenchmark_5epochs.py
index a76bb634c..f11e44e01 100644
--- a/nnunetv2/training/nnUNetTrainer/variants/benchmarking/nnUNetTrainerBenchmark_5epochs.py
+++ b/nnunetv2/training/nnUNetTrainer/variants/benchmarking/nnUNetTrainerBenchmark_5epochs.py
@@ -29,6 +29,7 @@ def run_training(self):
             super().run_training()
         except RuntimeError:
             self.crashed_with_runtime_error = True
+        self.on_train_end()
 
     def on_train_end(self):
         super().on_train_end()