diff --git a/kohya_gui/class_flux1.py b/kohya_gui/class_flux1.py
index f7af9424..452ae28d 100644
--- a/kohya_gui/class_flux1.py
+++ b/kohya_gui/class_flux1.py
@@ -108,7 +108,7 @@ def noise_offset_type_change(
                     )
                     self.timestep_sampling = gr.Dropdown(
                         label="Timestep Sampling",
-                        choices=["sigma", "shift", "sigmoid", "uniform"],
+                        choices=["flux_shift ", "sigma", "shift", "sigmoid", "uniform"],
                         value=self.config.get("flux1.timestep_sampling", "sigma"),
                         interactive=True,
                     )