diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 64391d9f5..38a35c731 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -65,6 +65,7 @@ def get_parser():
     )
     parser.add_argument(
         "--dist_rank",
+        type=int,
         default=None,
         help="node rank for distributed training",
     )