From ab65468a2c15839ac11f126bac3d372c11489be8 Mon Sep 17 00:00:00 2001
From: Hongze Zhang <hongze.zhang@intel.com>
Date: Fri, 10 Nov 2023 14:43:11 +0800
Subject: [PATCH] fixup

---
 .../memory/memtarget/spark/TreeMemoryConsumers.java             | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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());
         });
       }
     }