diff --git a/comfy/model_base.py b/comfy/model_base.py
index 9a6d84d6e..11624c6b0 100644
--- a/comfy/model_base.py
+++ b/comfy/model_base.py
@@ -1218,8 +1218,8 @@ class WAN21(BaseModel):
         print(f"image shape: {image.shape}, mask shape: {mask.shape}")
         res = torch.cat((mask, image), dim=1)
         tracks = kwargs.get("tracks", None)
-        # if tracks is not None:
-        #     res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
+        if tracks is not None:
+            res = patch_motion(tracks.to(device), res[0], 220.0, (4, 16), 2)[None]
         
         return res