diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/audit/AuditManager.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/audit/AuditManager.java
index 0230b38b..7e263e2e 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/audit/AuditManager.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/audit/AuditManager.java
@@ -110,6 +110,8 @@ public class AuditManager {
             T result = supplier.execute();
             if (result instanceof Collection) {
                 auditMessage.setQueryCount(((Collection) result).size());
+            } else if (result instanceof Integer) {
+                auditMessage.setQueryCount((int) result);
             } else if (result != null) {
                 auditMessage.setQueryCount(1);
             }