diff --git a/internal/datanode/importv2/util_test.go b/internal/datanode/importv2/util_test.go
index a7c7d27a86670..43698294c51b6 100644
--- a/internal/datanode/importv2/util_test.go
+++ b/internal/datanode/importv2/util_test.go
@@ -149,7 +149,7 @@ func Test_PickSegment(t *testing.T) {
 	importedSize := map[int64]int{}
 
 	totalSize := 8 * 1024 * 1024 * 1024
-	batchSize := 16 * 1024 * 1024
+	batchSize := 1 * 1024 * 1024
 
 	for totalSize > 0 {
 		picked := PickSegment(task, vchannel, partitionID)