diff --git a/tests/schedulers/test_schedulers.py b/tests/schedulers/test_schedulers.py
index 58acf42576..fc7f22d2a8 100755
--- a/tests/schedulers/test_schedulers.py
+++ b/tests/schedulers/test_schedulers.py
@@ -716,7 +716,7 @@ class SchedulerCommonTest(unittest.TestCase):
                 # Get valid timestep based on sigma_max, which should always be in timestep schedule.
                 scaled_sigma_max = scheduler.sigma_to_t(scheduler.config.sigma_max)
                 scaled_sample = scheduler.scale_model_input(sample, scaled_sigma_max)
-            if scheduler_class == EDMEulerScheduler:
+            elif scheduler_class == EDMEulerScheduler:
                 scaled_sample = scheduler.scale_model_input(sample, scheduler.timesteps[-1])
             else:
                 scaled_sample = scheduler.scale_model_input(sample, 0.0)