diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
index 59588cb9f..9d499bfae 100644
Binary files a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp differ
diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py
index b025383a5..68d095397 100755
--- a/egs/librispeech/ASR/conformer_ctc2/decode.py
+++ b/egs/librispeech/ASR/conformer_ctc2/decode.py
@@ -415,9 +415,10 @@ def decode_one_batch(
         return {key: hyps}
     
     if params.method == "greedy-search":
+        memory, memory_key_padding_mask
         att_loss = model.decoder_forward(
-            encoder_memory,
-            memory_mask,
+            memory,
+            momory_key_padding_mask,
             token_ids=unsorted_token_ids,
             sos_id=graph_compiler.sos_id,
             eos_id=graph_compiler.eos_id,