diff --git a/mu_map/training/default.py b/mu_map/training/default.py
index d0d627b663b1893fe0c7d7ad7c4c7de57ded6ba1..7f41f43fe8d98ebd83036575485f45dd1de8a708 100644
--- a/mu_map/training/default.py
+++ b/mu_map/training/default.py
@@ -40,7 +40,7 @@ class Training():
             self.lr_scheduler.step()
             logger.debug(f"Update learning rate from {_previous:.4f} to {self.lr_scheduler.get_last_lr()[0]:.4f}")
 
-            if epoch % self.snapshot_epoch:
+            if epoch % self.snapshot_epoch == 0:
                 self.store_snapshot(epoch)
 
             logger.debug(f"Finished epoch {str(epoch):>{len(str(self.epochs))}}/{self.epochs + 1}")