diff --git a/wenet/bin/export_jit.py b/wenet/bin/export_jit.py
index 70f53586b..98eadd61c 100644
--- a/wenet/bin/export_jit.py
+++ b/wenet/bin/export_jit.py
@@ -38,6 +38,7 @@ def get_args():
 
 def main():
     args = get_args()
+    args.jit = True
     logging.basicConfig(level=logging.DEBUG,
                         format='%(asctime)s %(levelname)s %(message)s')
     # No need gpu for model export