diff --git a/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java
index 072c4bc8f0e1..5d0ed0beadca 100644
--- a/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java
+++ b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java
@@ -1508,16 +1508,11 @@ private static List<String> buildRowsForInsert(int numberOfRows)
     public static Object[][] batchSizeAndTotalNumberOfRowsToInsertDataProvider()
     {
         return new Object[][] {
-                {100, 64},
-                {100, 100},
-                {100, 512},
-                {100, 1000},
-                {1000, 100},
-                {1000, 1000},
-                {1000, 5000},
-                {10000, 1000},
-                {10000, 5000},
-                {10000, 15000},
+                {10, 8},  // number of rows < batch size
+                {10, 10}, // number of rows = batch size
+                {10, 11}, // number of rows > batch size
+                {10, 50}, // number of rows = n * batch size
+                {10, 52}, // number of rows > n * batch size
         };
     }
 }