diff --git a/onnxruntime/python/tools/transformers/fusion_attention.py b/onnxruntime/python/tools/transformers/fusion_attention.py
index 23679bfbe0..b6d36d6e08 100644
--- a/onnxruntime/python/tools/transformers/fusion_attention.py
+++ b/onnxruntime/python/tools/transformers/fusion_attention.py
@@ -345,7 +345,7 @@ class FusionAttention(Fusion):
         )
         einsum_node = None
         if qkv_nodes is not None:
-            (_, matmul_qkv, reshape_qkv, transpose_qkv, matmul_qkv) = qkv_nodes
+            (_, _, reshape_qkv, transpose_qkv, matmul_qkv) = qkv_nodes
         else:
             # Match Albert
             qkv_nodes = self.model.match_parent_path(