diff --git a/tilelang/transform/z3_scheduler.py b/tilelang/transform/z3_scheduler.py
index 9fb18c7da6..4cd967fe2d 100644
--- a/tilelang/transform/z3_scheduler.py
+++ b/tilelang/transform/z3_scheduler.py
@@ -462,7 +462,7 @@ def z3_schedule_loop_python(
                 k_vars.append(k_var)
                 r_vars.append(r_var)
             offset = min(k_vars)
-        begin = begin - offset * ii_mid
+        begin = begin - offset * best_ii
         for i in range(n):
             start_time = best_model.eval(best_start_vars[i]).as_long() - offset * best_ii
             stage = best_model.eval(best_stage_vars[i]).as_long()