diff --git a/src/metatrain/experimental/pet/trainer.py b/src/metatrain/experimental/pet/trainer.py
index 73bbd4a9c..39878fbd4 100644
--- a/src/metatrain/experimental/pet/trainer.py
+++ b/src/metatrain/experimental/pet/trainer.py
@@ -719,7 +719,7 @@ def save_model(model_name, model_keeper):
             os.remove(Path(checkpoint_dir) / "checkpoint.temp")
 
         if do_forces:
-            load_path = self.pet_dir / "best_val_mae_forces_model_state_dict"
+            load_path = self.pet_dir / "best_val_mae_both_model_state_dict"
         else:
             load_path = self.pet_dir / "best_val_mae_energies_model_state_dict"