diff --git a/tests/pytest/insert/boundary2.py b/tests/pytest/insert/boundary2.py
index 99784b7cd9cfa7fc8d32720442cefa7946f6511f..8a6fd1e6a1060c6bfd5f8ec5c57a5d8aef4922bd 100644
--- a/tests/pytest/insert/boundary2.py
+++ b/tests/pytest/insert/boundary2.py
@@ -50,14 +50,16 @@ class TDTestCase:
             sql += "'%s')" % self.get_random_string(22)
             tdSql.execute(sql % (self.ts + i))
 
-        tdSql.query("select * from stb")
-        tdSql.checkRows(4096)
+        time.sleep(10)
+        tdSql.query("select count(*) from stb")
+        tdSql.checkData(0, 0, 4096)
 
         tdDnodes.stop(1)
         tdDnodes.start(1)
-        
-        tdSql.query("select * from stb")
-        tdSql.checkRows(4096)
+
+        time.sleep(1)
+        tdSql.query("select count(*) from stb")
+        tdSql.checkData(0, 0, 4096)
 
         endTime = time.time()