diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
index 67d6febf7..b90dedbf0 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 0ab55ef43..f1a56fb05 100755
--- a/egs/librispeech/ASR/conformer_ctc2/decode.py
+++ b/egs/librispeech/ASR/conformer_ctc2/decode.py
@@ -319,8 +319,8 @@ def decode_one_batch(
     supervisions = batch["supervisions"]
 
     nnet_output, memory, memory_key_padding_mask = model(feature, supervisions)
-    print(type(nnet_output))
-    exit()
+    if type(nnet_output) == tuple:
+        nnet_output = nnet_output[0]
     # nnet_output is (N, T, C)
 
     supervision_segments = torch.stack(