diff --git a/gluten-core/src/main/java/io/glutenproject/memory/memtarget/spark/TreeMemoryConsumers.java b/gluten-core/src/main/java/io/glutenproject/memory/memtarget/spark/TreeMemoryConsumers.java
index 172be6e5743e..317e20cda7b4 100644
--- a/gluten-core/src/main/java/io/glutenproject/memory/memtarget/spark/TreeMemoryConsumers.java
+++ b/gluten-core/src/main/java/io/glutenproject/memory/memtarget/spark/TreeMemoryConsumers.java
@@ -56,7 +56,7 @@ private TreeMemoryTarget getSharedAccount(TaskMemoryManager tmm) {
       synchronized (MAP) {
         return MAP.computeIfAbsent(tmm, m -> {
           TreeMemoryTarget tmc = new TreeMemoryConsumer(m);
-          return tmc.newChild("root", GlutenConfig.getConf().conservativeTaskOffHeapMemorySize(), Spiller.NO_OP, Collections.emptyMap());
+          return tmc.newChild("root", perTaskCapacity, Spiller.NO_OP, Collections.emptyMap());
         });
       }
     }