diff --git a/mu_map/training/lib.py b/mu_map/training/lib.py
index b43f1bfd619311671a192bc6fdef021b32fb1e0e..eb23b5741b0d6cdf910c33c4bdaf89130e785bea 100644
--- a/mu_map/training/lib.py
+++ b/mu_map/training/lib.py
@@ -78,7 +78,7 @@ class AbstractTraining:
                 self.store_snapshot("val_min")
 
             if epoch % self.snapshot_epoch == 0:
-                self._store_snapshot(f"{epoch:0d{len(str(self.epochs))}}")
+                self.store_snapshot(f"{epoch:0d{len(str(self.epochs))}}")
 
             for param in self.training_params:
                 if param.lr_scheduler is not None: