diff --git a/hudi-flink-datasource/hudi-flink/src/test/java/org/apache/hudi/sink/ITTestDataStreamWrite.java b/hudi-flink-datasource/hudi-flink/src/test/java/org/apache/hudi/sink/ITTestDataStreamWrite.java
index b9deb43a97c5..9c5eeeeb5a82 100644
--- a/hudi-flink-datasource/hudi-flink/src/test/java/org/apache/hudi/sink/ITTestDataStreamWrite.java
+++ b/hudi-flink-datasource/hudi-flink/src/test/java/org/apache/hudi/sink/ITTestDataStreamWrite.java
@@ -102,7 +102,7 @@ public void testWriteCopyOnWrite(String indexType) throws Exception {
     conf.setString(FlinkOptions.INDEX_KEY_FIELD, "id");
     conf.setBoolean(FlinkOptions.PRE_COMBINE,true);
 
-    testWriteToHoodie(conf, "cow_write", 1, EXPECTED);
+    testWriteToHoodie(conf, "cow_write", 2, EXPECTED);
   }
 
   @Test