diff --git a/comfy/model_base.py b/comfy/model_base.py
index f155f8096..9a6d84d6e 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1221,7 +1221,7 @@ class WAN21(BaseModel):
         # if tracks is not None:
         #     res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
         
-        return res[None]
+        return res
 
 
     def extra_conds(self, **kwargs):