diff --git a/cpp/velox/jni/VeloxJniWrapper.cc b/cpp/velox/jni/VeloxJniWrapper.cc
index a0889cf5a952..5633084c0446 100644
--- a/cpp/velox/jni/VeloxJniWrapper.cc
+++ b/cpp/velox/jni/VeloxJniWrapper.cc
@@ -581,7 +581,7 @@ Java_org_apache_gluten_datasource_VeloxDataSourceJniWrapper_splitBlockByPartitio
   const auto numRows = inputRowVector->size();
 
   connector::hive::PartitionIdGenerator idGen(
-      asRowType(inputRowVector->type()), partitionColIndicesVec, 128, pool.get()
+      asRowType(inputRowVector->type()), partitionColIndicesVec, 65536, pool.get()
 #ifdef GLUTEN_ENABLE_ENHANCED_FEATURES
       ,
       true