diff --git a/nodes.py b/nodes.py
index c5c8c2c..3e15b0e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3018,7 +3018,7 @@ class WanVideoSampler:
 
                     partial_unianim_data = None
                     if unianim_data is not None:
-                        partial_dwpose = dwpose_data[:, c, :, :]
+                        partial_dwpose = dwpose_data[:, :, c, :, :]
                         partial_dwpose_flat=rearrange(partial_dwpose, 'b c f h w -> b (f h w) c')
                         partial_unianim_data = {
                             "dwpose": partial_dwpose_flat,