diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
index 74ece2527..cb618e180 100644
--- a/scripts/xyz_grid.py
+++ b/scripts/xyz_grid.py
@@ -220,10 +220,10 @@ axis_options = [
     AxisOption("Sigma min", float, apply_field("s_tmin")),
     AxisOption("Sigma max", float, apply_field("s_tmax")),
     AxisOption("Sigma noise", float, apply_field("s_noise")),
-    AxisOption("KDiffusion Scheduler Type", str, apply_field("k_sched_type"), choices=lambda: [x for x in sd_samplers_kdiffusion.k_diffusion_scheduler]),
-    AxisOption("KDiffusion Scheduler Sigma Min", float, apply_field("sigma_min")),
-    AxisOption("KDiffusion Scheduler Sigma Max", float, apply_field("sigma_max")),
-    AxisOption("KDiffusion Scheduler rho", float, apply_field("rho")),
+    AxisOption("KDiffusion Scheduler Type", str, apply_override("k_sched_type"), choices=lambda: [x for x in sd_samplers_kdiffusion.k_diffusion_scheduler]),
+    AxisOption("KDiffusion Scheduler Sigma Min", float, apply_override("sigma_min")),
+    AxisOption("KDiffusion Scheduler Sigma Max", float, apply_override("sigma_max")),
+    AxisOption("KDiffusion Scheduler rho", float, apply_override("rho")),
     AxisOption("Eta", float, apply_field("eta")),
     AxisOption("Clip skip", int, apply_clip_skip),
     AxisOption("Denoising", float, apply_field("denoising_strength")),