diff --git a/nodes_rf_inversion.py b/nodes_rf_inversion.py
index 170f2f8..ab99ef4 100644
--- a/nodes_rf_inversion.py
+++ b/nodes_rf_inversion.py
@@ -252,7 +252,7 @@ class HyVideoInverseSampler:
                 
                 progress_bar.update()
                 if callback is not None:
-                        callback(idx, latents.detach()[-1].permute(1,0,2,3), None, steps)
+                    callback(idx, (latent_model_input - noise_pred).detach()[0].permute(1,0,2,3), None, steps)
                 else:
                     comfy_pbar.update(1)
                   
@@ -458,7 +458,7 @@ class HyVideoReSampler:
 
                     progress_bar.update()
                     if callback is not None:
-                        callback(idx, latents.detach()[-1].permute(1,0,2,3), None, steps)
+                        callback(idx, (latent_model_input - noise_pred).detach()[0].permute(1,0,2,3), None, steps)
                     else:
                         comfy_pbar.update(1)
 
@@ -695,7 +695,7 @@ class HyVideoPromptMixSampler:
 
                     progress_bar.update()
                     if callback is not None:
-                        callback(idx, latents.detach()[-1].permute(1,0,2,3), None, steps)
+                        callback(idx, (latent_model_input - noise_pred).detach()[0].permute(1,0,2,3), None, steps)
                     else:
                         comfy_pbar.update(1)