diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
index 98d6fe937..a9159bd88 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 3eeedb424..d96f1bcf9 100755
--- a/egs/librispeech/ASR/conformer_ctc2/decode.py
+++ b/egs/librispeech/ASR/conformer_ctc2/decode.py
@@ -424,6 +424,7 @@ def decode_one_batch(
             sos_id=graph_compiler.sos_id,
             eos_id=graph_compiler.eos_id,
             warmup=warmup,
+            return_output=True,
         )
         hyps = greedy_search(nnet_output, memory_key_padding_mask)