diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp
index 1bc205f83..5571c93cb 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
index 13c20a989..71a2fee14 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
@@ -496,9 +496,9 @@ class Data2VecAudioModel(BaseFairseqModel):
         
         print(padding_mask.size())
         print((padding_mask[0] == True).nonzero(as_tuple=True)[0])
-        print((padding_mask[1] == True).nonzero(as_tuple=True)[0])
-        print((padding_mask[2] == True).nonzero(as_tuple=True)[0])
-        print((padding_mask[3] == True).nonzero(as_tuple=True)[0])
+        print((padding_mask[1] == True).nonzero(as_tuple=True)[0][1])
+        print((padding_mask[2] == True).nonzero(as_tuple=True)[0][2])
+        print((padding_mask[3] == True).nonzero(as_tuple=True)[0][3])
         exit()
 
         if self.post_extract_proj is not None: