diff --git a/source/libs/parser/inc/sql.y b/source/libs/parser/inc/sql.y
index be425107c8..4ed05d6b74 100644
--- a/source/libs/parser/inc/sql.y
+++ b/source/libs/parser/inc/sql.y
@@ -859,9 +859,9 @@ cmd ::= KILL TRANSACTION NK_INTEGER(A).
 cmd ::= KILL COMPACT NK_INTEGER(A).                                               { pCxt->pRootNode = createKillStmt(pCxt, QUERY_NODE_KILL_COMPACT_STMT, &A); }
 
 /************************************************ merge/redistribute/ vgroup ******************************************/
-cmd ::= BALANCE VGROUP.                                                           { pCxt->pRootNode = createAssignLeaderStmt(pCxt); }
+cmd ::= BALANCE VGROUP.                                                           { pCxt->pRootNode = createBalanceVgroupStmt(pCxt); }
 
-cmd ::= ASSIGN LEADER FORCE.                                                      { pCxt->pRootNode = createAssignLeaderStmt(pCxt); }
+cmd ::= BALANCE VGROUP FORCE.                                                      { pCxt->pRootNode = createAssignLeaderStmt(pCxt); }
 
 cmd ::= BALANCE VGROUP LEADER on_vgroup_id(A).                                    { pCxt->pRootNode = createBalanceVgroupLeaderStmt(pCxt, &A); }
 cmd ::= BALANCE VGROUP LEADER DATABASE db_name(A).                                { pCxt->pRootNode = createBalanceVgroupLeaderDBNameStmt(pCxt, &A); }
diff --git a/tests/army/cluster/arbitrator_restart.py b/tests/army/cluster/arbitrator_restart.py
index 7a02e46630..21b7a5b82b 100644
--- a/tests/army/cluster/arbitrator_restart.py
+++ b/tests/army/cluster/arbitrator_restart.py
@@ -71,7 +71,7 @@ class TDTestCase(TBase):
 
             count += 1
 
-        tdSql.execute("balance vgroup;")
+        tdSql.execute("BALANCE VGROUP FORCE;")
 
         count = 0
         while count < 100:
diff --git a/tests/parallel_test/cases.task b/tests/parallel_test/cases.task
index 0201c88d2b..a38b0cd45e 100644
--- a/tests/parallel_test/cases.task
+++ b/tests/parallel_test/cases.task
@@ -22,6 +22,7 @@
 ,,y,army,./pytest.sh python3 ./test.py -f multi-level/mlevel_basic.py -N 3 -L 3 -D 2
 ,,y,army,./pytest.sh python3 ./test.py -f db-encrypt/basic.py -N 3 -M 3
 ,,y,army,./pytest.sh python3 ./test.py -f cluster/arbitrator.py -N 3
+,,y,army,./pytest.sh python3 ./test.py -f cluster/arbitrator_restart.py -N 3
 ,,n,army,python3 ./test.py -f storage/s3/s3Basic.py -N 3
 ,,y,army,./pytest.sh python3 ./test.py -f cluster/snapshot.py -N 3 -L 3 -D 2
 ,,y,army,./pytest.sh python3 ./test.py -f query/function/test_func_elapsed.py