diff --git a/services/ans/test/unittest/notification_rdb_data_mgr_test/notification_rdb_data_mgr_test.cpp b/services/ans/test/unittest/notification_rdb_data_mgr_test/notification_rdb_data_mgr_test.cpp
index 4f8469b7e3347a08f2c9d688a4ba1d603b1f6a30..f1aa458be60f1471be5829ed53d146efd4e7384b 100755
--- a/services/ans/test/unittest/notification_rdb_data_mgr_test/notification_rdb_data_mgr_test.cpp
+++ b/services/ans/test/unittest/notification_rdb_data_mgr_test/notification_rdb_data_mgr_test.cpp
@@ -127,7 +127,7 @@ class RdbStoreTest : public RdbStore {
             return nullptr;
         };
         virtual std::shared_ptr<ResultSet> QueryByStep(
-            const std::string &sql, const std::vector<ValueObject> &bindArgs = {})
+            const std::string &sql, const std::vector<ValueObject> &bindArgs = {}, bool preCount = true)
         {
             return nullptr;
         };
@@ -185,7 +185,7 @@ class RdbStoreTest : public RdbStore {
             return nullptr;
         };
         virtual std::shared_ptr<ResultSet> QueryByStep(
-            const AbsRdbPredicates &predicates, const std::vector<std::string> &columns)
+            const AbsRdbPredicates &predicates, const std::vector<std::string> &columns, bool preCount = true)
         {
             return nullptr;
         };