diff --git a/modules/shared_state.py b/modules/shared_state.py
index 699cb0fc..6ea8e961 100644
--- a/modules/shared_state.py
+++ b/modules/shared_state.py
@@ -172,6 +172,8 @@ class State:
                 else:
                     self.assign_current_image(modules.sd_samplers.sample_to_image(self.current_latent))
 
+            self.current_image_sampling_step = self.sampling_step
+
         except Exception as e:
             # traceback.print_exc()
             # print(e)