diff --git a/nodes.py b/nodes.py
index d836856..e505996 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3174,7 +3174,7 @@ class WanVideoSampler:
             "drop_last": drop_last,
             "generator_state": seed_g.get_state(),
         },{
-            "samples": callback_latent.unsqueeze(0).cpu(), 
+            "samples": callback_latent.unsqueeze(0).cpu() if callback is not None else None, 
         })
 
 #region VideoDecode