diff --git a/restart_sampling.py b/restart_sampling.py
index a296f12..d2b972a 100644
--- a/restart_sampling.py
+++ b/restart_sampling.py
@@ -469,7 +469,7 @@ class RestartPlan:
             restart_sigmas = calc_sigmas(
                 restart_scheduler,
                 n_restart,
-                max(model_sigma_min, sigmas[i + 1]),
+                max(model_sigma_min, s_min),
                 s_max,
                 model,
                 device=device,