diff --git a/src/careamics_napari/widgets/train_progress_widget.py b/src/careamics_napari/widgets/train_progress_widget.py
index 7483317..d2a6c1e 100644
--- a/src/careamics_napari/widgets/train_progress_widget.py
+++ b/src/careamics_napari/widgets/train_progress_widget.py
@@ -91,7 +91,7 @@ def _update_training_state(self: Self, state: TrainingState) -> None:
             Training state.
         """
         if state == TrainingState.IDLE or state == TrainingState.TRAINING:
-            self.plot.clear()
+            self.plot.clear_plot()
 
     def _update_max_epoch(self: Self, max_epoch: int):
         """Update the maximum number of epochs in the progress bar.