diff --git a/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp b/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp
index b0fbe45f470e..5ea6a5715266 100644
--- a/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp
+++ b/velox/exec/tests/ScaleWriterLocalPartitionTest.cpp
@@ -552,7 +552,7 @@ class ScaleWriterLocalPartitionTest : public HiveConnectorTestBase {
 };
 
 TEST_F(ScaleWriterLocalPartitionTest, unpartitionBasic) {
-  const std::vector<RowVectorPtr> inputVectors = makeVectors(32, 1024);
+  const std::vector<RowVectorPtr> inputVectors = makeVectors(128, 1024);
   const uint64_t queryCapacity = 256 << 20;
   const uint32_t maxDrivers = 32;
   const uint32_t maxExchanegBufferSize = 2 << 20;