diff --git a/setup.py b/setup.py
index 21c51ce3c..6189a650b 100644
--- a/setup.py
+++ b/setup.py
@@ -142,6 +142,7 @@ setup(
         "console_scripts": [
             "funasr = funasr.bin.inference:main_hydra",
             "funasr-train = funasr.bin.train:main_hydra",
+            "funasr-train-ds = funasr.bin.train_ds:main_hydra",
             "funasr-export = funasr.bin.export:main_hydra",
             "scp2jsonl = funasr.datasets.audio_datasets.scp2jsonl:main_hydra",
             "jsonl2scp = funasr.datasets.audio_datasets.jsonl2scp:main_hydra",