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
new file mode 100644
index 000000000..01d980981
Binary files /dev/null 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 67e8506be..73653ff63 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
@@ -495,7 +495,7 @@ class Data2VecAudioModel(BaseFairseqModel):
             padding_mask = None
         
         print(padding_mask.size())
-        print(len((padding_mask == True).nonzero(as_tuple=True)))
+        print(len((padding_mask == True).nonzero(as_tuple=True)[0]))
         exit()
 
         if self.post_extract_proj is not None: