diff --git a/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py b/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py
index c0815a6..1b56138 100644
--- a/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py
+++ b/src/poli_baselines/solvers/bayesian_optimization/lambo2/solver.py
@@ -213,6 +213,7 @@ def _train_model_with_history(
                 torch.load(
                     load_checkpoint_from,
                     map_location="cpu",
+                    weights_only=False,
                 )["state_dict"]
             )