diff --git a/wanvideo/modules/attention_flash.py b/wanvideo/modules/attention_flash.py
index ed8e99b..bddeb37 100644
--- a/wanvideo/modules/attention_flash.py
+++ b/wanvideo/modules/attention_flash.py
@@ -66,7 +66,7 @@ else:
                     0, dtype=torch.int32).to(q.device, non_blocking=True),
                 seqused_q=None, seqused_k=None, max_seqlen_q=lq, max_seqlen_k=lk,
                 softmax_scale=softmax_scale, causal=causal,
-                deterministic=deterministic)[0].unflatten(0, (b, lq))
+                deterministic=deterministic).unflatten(0, (b, lq))
         else:
             assert FLASH_ATTN_2_AVAILABLE
             x = flash_attn.flash_attn_varlen_func(q=q, k=k, v=v,