diff --git a/test/test_decode.py b/test/test_decode.py
index ca89c6e02..7ef127781 100644
--- a/test/test_decode.py
+++ b/test/test_decode.py
@@ -23,7 +23,7 @@ You can run this file in one of the two ways:
 """
 
 import k2
-from icefall.decode2 import Nbest
+from icefall.decode import Nbest
 
 
 def test_nbest_from_lattice():