diff --git a/imaginairy/api/generate_refiners.py b/imaginairy/api/generate_refiners.py
index f20546c1..6f297370 100644
--- a/imaginairy/api/generate_refiners.py
+++ b/imaginairy/api/generate_refiners.py
@@ -306,7 +306,7 @@ def latent_logger(latents):
 
         if init_latent is not None:
             noise_step = noise_step if noise_step is not None else first_step
-            if first_step >= len(sd.steps):
+            if first_step >= len(sd.scheduler.all_steps):
                 noised_latent = init_latent
             else:
                 noised_latent = sd.scheduler.add_noise(
diff --git a/tests/expected_output/test_large_image_.png b/tests/expected_output/test_large_image_.png
index 517e4a10..d7cac480 100644
Binary files a/tests/expected_output/test_large_image_.png and b/tests/expected_output/test_large_image_.png differ