diff --git a/HBI/models/banzhaf.py b/HBI/models/banzhaf.py
index 7b9aa24..4ea1a3d 100644
--- a/HBI/models/banzhaf.py
+++ b/HBI/models/banzhaf.py
@@ -126,6 +126,7 @@ def banzhaf_interaction(self, retrieve_logits, text_mask, video_mask, text_weigh
 
         #########################
         _text_mask[:, i] = 1
+        _video_mask[:, j] = 0
 
         _text_weight0, _video_weight0 = text_weight.clone(), video_weight.clone()
         _retrieve_logits0 = retrieve_logits.clone()
@@ -146,6 +147,7 @@ def banzhaf_interaction(self, retrieve_logits, text_mask, video_mask, text_weigh
         banzhaf_value2 = (t2v_logits + v2t_logits) / 2.0
 
         #########################
+        _text_mask[:, i] = 0
         _video_mask[:, j] = 1
 
         _text_weight0, _video_weight0 = text_weight.clone(), video_weight.clone()