diff --git a/icefall/.decode.py.swp b/icefall/.decode.py.swp
index 40d12e8c9..3bb27d68f 100644
Binary files a/icefall/.decode.py.swp and b/icefall/.decode.py.swp differ
diff --git a/icefall/decode.py b/icefall/decode.py
index e4c614c4e..8d24b734f 100644
--- a/icefall/decode.py
+++ b/icefall/decode.py
@@ -138,6 +138,8 @@ def get_lattice(
     Returns:
       An FsaVec containing the decoding result. It has axes [utt][state][arc].
     """
+    if type(nnet_output) == tuple:
+        nnet_output = nnet_output[0]
     dense_fsa_vec = k2.DenseFsaVec(
         nnet_output,
         supervision_segments,