diff --git a/tests/army/community/cluster/snapshot.py b/tests/army/community/cluster/snapshot.py
index 8543101300..8d15530061 100644
--- a/tests/army/community/cluster/snapshot.py
+++ b/tests/army/community/cluster/snapshot.py
@@ -72,8 +72,8 @@ class TDTestCase(TBase):
 
         # check count always return value
         sql = f"select count(*) from {self.db}.ta"
-        tdSql.waitedQuery(sql, 0, 2)        
-
+        tdSql.query()
+        tdSql.checkRows(0) # countAlwaysReturnValue is false
 
     # run
     def run(self):
diff --git a/tests/army/frame/caseBase.py b/tests/army/frame/caseBase.py
index ca5539f75b..aca2b524da 100644
--- a/tests/army/frame/caseBase.py
+++ b/tests/army/frame/caseBase.py
@@ -132,8 +132,9 @@ class TBase:
         tdSql.checkAgg(sql, self.childtable_count)
 
         # check step
-        sql = f"select count(*) from (select diff(ts) as dif from {self.stb} partition by tbname) where dif != {self.timestamp_step}"
-        tdSql.checkAgg(sql, 0)
+        sql = f"select * from (select diff(ts) as dif from {self.stb} partition by tbname) where dif != {self.timestamp_step}"
+        tdSql.query(sql)
+        tdSql.checkRows(0)
 
     # save agg result
     def snapshotAgg(self):