diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp
index d6ac48fae..e62d6b83c 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
index 107e06455..51f034fab 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
@@ -52,7 +52,6 @@ class FairSeqData2VecEncoder(EncoderInterface):
         output_size: int = 256,
         freeze_finetune_updates: int = 0,
         additional_block: bool = False,
-        prompt = None,
     ):
         assert check_argument_types()
         super().__init__()
@@ -97,6 +96,7 @@ class FairSeqData2VecEncoder(EncoderInterface):
         x_lens: torch.Tensor,
         warmup = None,
         prev_states: torch.Tensor = None,
+        prompt: torch.Tensor = None,
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         xs_pad = x
         ilens = x_lens