diff --git a/mybatis-flex-core/src/main/java/com/mybatisflex/core/row/RowMapperInvoker.java b/mybatis-flex-core/src/main/java/com/mybatisflex/core/row/RowMapperInvoker.java
index a6279056..45f457a6 100644
--- a/mybatis-flex-core/src/main/java/com/mybatisflex/core/row/RowMapperInvoker.java
+++ b/mybatis-flex-core/src/main/java/com/mybatisflex/core/row/RowMapperInvoker.java
@@ -101,7 +101,7 @@ public class RowMapperInvoker {
             RowMapper mapper = sqlSession.getMapper(RowMapper.class);
             int counter = 0;
             int resultsPos = 0;
-            for (int i = 0; i < batchSize; i++) {
+            for (int i = 0; i < totalSize; i++) {
                 if (++counter > batchSize) {
                     counter = 0;
                     List<BatchResult> batchResults = sqlSession.flushStatements();