diff --git a/src/prime_rl/orchestrator/ckpt.py b/src/prime_rl/orchestrator/ckpt.py
index 19277e3063..b98123b61c 100644
--- a/src/prime_rl/orchestrator/ckpt.py
+++ b/src/prime_rl/orchestrator/ckpt.py
@@ -57,7 +57,7 @@ def load_from_path(self, ckpt_path: Path, progress: Progress, buffer: Buffer) ->
             self.logger.info("Skipping progress loading from checkpoint")
         else:
             with open(ckpt_path / "progress.pt", "rb") as f:
-                state = torch.load(f, weights_only=False)
+                state = torch.load(f, weights_only=True)
 
             # Set progress in-place
             for key, value in asdict(state["progress"]).items():