diff --git a/tests/parallel_test/cases.task b/tests/parallel_test/cases.task
index 40eb926efb..7822e9614a 100644
--- a/tests/parallel_test/cases.task
+++ b/tests/parallel_test/cases.task
@@ -20,9 +20,9 @@
 ,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/window_close_session_ext.py
 ,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/partition_interval.py
 ,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/pause_resume_test.py
-#,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/vnode_restart.py
-#,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/snode_restart.py
-,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/snode_restart_with_checkpoint.py
+#,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/vnode_restart.py -N 4
+#,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/snode_restart.py -N 4
+,,y,system-test,./pytest.sh python3 ./test.py -f 8-stream/snode_restart_with_checkpoint.py -N 4
 
 ,,y,system-test,./pytest.sh python3 ./test.py -f 2-query/tbname_vgroup.py
 ,,y,system-test,./pytest.sh python3 ./test.py -f 2-query/stbJoin.py
diff --git a/tests/system-test/2-query/db.py b/tests/system-test/2-query/db.py
index 6870c59a0d..0246626e40 100644
--- a/tests/system-test/2-query/db.py
+++ b/tests/system-test/2-query/db.py
@@ -55,7 +55,7 @@ class TDTestCase:
             tdSql.checkData(0, 2, 0)
 
         tdSql.query("show dnode 1 variables like '%debugFlag'")
-        tdSql.checkRows(22)
+        tdSql.checkRows(23)
 
         tdSql.query("show dnode 1 variables like '____debugFlag'")
         tdSql.checkRows(2)
diff --git a/tests/system-test/7-tmq/tmqVnodeReplicate.py b/tests/system-test/7-tmq/tmqVnodeReplicate.py
index afb3319491..0ee11781ed 100644
--- a/tests/system-test/7-tmq/tmqVnodeReplicate.py
+++ b/tests/system-test/7-tmq/tmqVnodeReplicate.py
@@ -132,14 +132,15 @@ class TDTestCase:
         tmqCom.getStartConsumeNotifyFromTmqsim()
         tmqCom.getStartCommitNotifyFromTmqsim()
 
-        tdSql.query("select * from information_schema.ins_vnodes")
-        # tdLog.debug(tdSql.queryResult)
-        tdDnodes = cluster.dnodes
-        for result in tdSql.queryResult:
-            if result[2] == 'dbt' and result[3] == 'leader':
-                tdLog.debug("leader is %d"%(result[0] - 1))
-                tdDnodes[result[0] - 1].stoptaosd()
-                break
+        tdSql.query("balance vgroup leader")
+        # tdSql.query("select * from information_schema.ins_vnodes")
+        # # tdLog.debug(tdSql.queryResult)
+        # tdDnodes = cluster.dnodes
+        # for result in tdSql.queryResult:
+        #     if result[2] == 'dbt' and result[3] == 'leader':
+        #         tdLog.debug("leader is %d"%(result[0] - 1))
+        #         tdDnodes[result[0] - 1].stoptaosd()
+        #         break
 
         pInsertThread.join()
         expectRows = 1
@@ -158,7 +159,6 @@ class TDTestCase:
         tdLog.printNoPrefix("======== test case 1 end ...... ")
 
     def run(self):
-        tdSql.prepare()
         self.prepareTestEnv()
         self.tmqCase1()