diff --git a/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py b/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py
index ce73ac5d..fc3e67e3 100644
--- a/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py
+++ b/src/maxdiffusion/models/wan/transformers/transformer_wan_vace.py
@@ -474,6 +474,7 @@ def __call__(
         timestep_proj,
         encoder_hidden_states,
         encoder_hidden_states_image,
+        _,
     ) = self.condition_embedder(  # We will need to mask out the text embedding.
         timestep, encoder_hidden_states, encoder_hidden_states_image
     )