diff --git a/icefall/decode.py b/icefall/decode.py
index 5f90ee168..5d836bd48 100644
--- a/icefall/decode.py
+++ b/icefall/decode.py
@@ -24,8 +24,8 @@ import k2
 import torch
 
 from icefall.context_graph import ContextGraph, ContextState
-from icefall.ngram_lm import NgramLm, NgramLmStateCost
 from icefall.lm_wrapper import LmScorer
+from icefall.ngram_lm import NgramLm, NgramLmStateCost
 from icefall.utils import add_eos, add_sos, get_texts
 
 DEFAULT_LM_SCALE = [