diff --git a/k_diffusion/sampling.py b/k_diffusion/sampling.py
index 9f10e8d..69e3294 100644
--- a/k_diffusion/sampling.py
+++ b/k_diffusion/sampling.py
@@ -646,9 +646,9 @@ def sample_dpmpp_2m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
 
             if eta:
                 x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * sigmas[i + 1] * (-2 * eta_h).expm1().neg().sqrt() * s_noise
-
+            h_last = h
+            
         old_denoised = denoised
-        h_last = h
     return x
 
 
@@ -696,7 +696,7 @@ def sample_dpmpp_3m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
 
             if eta:
                 x = x + noise_sampler(sigmas[i], sigmas[i + 1]) * sigmas[i + 1] * (-2 * h * eta).expm1().neg().sqrt() * s_noise
-
+            h_1, h_2 = h, h_1
+            
         denoised_1, denoised_2 = denoised, denoised_1
-        h_1, h_2 = h, h_1
     return x