diff --git a/tests/pytest/import_merge/importDataLastS.py b/tests/pytest/import_merge/importDataLastS.py
index 2c1559d2903a4fde104603bb2ad5945ea8eddd1e..4e5701deb1785819f915d9a51cbf4b9b5fc56bab 100644
--- a/tests/pytest/import_merge/importDataLastS.py
+++ b/tests/pytest/import_merge/importDataLastS.py
@@ -54,8 +54,8 @@ class TDTestCase:
         tdSql.execute(" ".join(sqlcmd))
 
         tdLog.info("================= step3")
-        tdSql.query('select * from tb1')
-        tdSql.checkRows(205)
+        tdSql.query('select count(*) from tb1')
+        tdSql.checkData(0, 0, 205)
 
         tdLog.info("================= step4")
         tdDnodes.stop(1)
@@ -71,8 +71,8 @@ class TDTestCase:
         tdSql.execute(" ".join(sqlcmd))
 
         tdLog.info("================= step6")
-        tdSql.query('select * from tb1')
-        tdSql.checkRows(250)
+        tdSql.query('select count(*) from tb1')
+        tdSql.checkData(0, 0, 250)
 
     def stop(self):
         tdSql.close()
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()