diff --git a/services/bundlemgr/test/mock/include/mock_app_provision_info.h b/services/bundlemgr/test/mock/include/mock_app_provision_info.h
index 3d46fe20a8c3400a1de2a91b14ffb1104c285e10..d7884f199774b782844b0d4c4b56a7d77231125b 100644
--- a/services/bundlemgr/test/mock/include/mock_app_provision_info.h
+++ b/services/bundlemgr/test/mock/include/mock_app_provision_info.h
@@ -102,7 +102,7 @@ public:
         return nullptr;
     };
     virtual std::shared_ptr<NativeRdb::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;
     };
@@ -153,7 +153,7 @@ public:
         return sharedResultSet;
     };
     virtual std::shared_ptr<NativeRdb::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;
     };