diff --git a/gluten-core/src/main/java/org/apache/gluten/memory/memtarget/TreeMemoryTargets.java b/gluten-core/src/main/java/org/apache/gluten/memory/memtarget/TreeMemoryTargets.java
index 98f79bfff367..26c6ea48008a 100644
--- a/gluten-core/src/main/java/org/apache/gluten/memory/memtarget/TreeMemoryTargets.java
+++ b/gluten-core/src/main/java/org/apache/gluten/memory/memtarget/TreeMemoryTargets.java
@@ -70,7 +70,7 @@ private static long spillTree(TreeMemoryTarget node, Spiller.Phase phase, final
     long remainingBytes = bytes;
     while (q.peek() != null && remainingBytes > 0) {
       TreeMemoryTarget head = q.remove();
-      long spilled = spillTree(head, remainingBytes);
+      long spilled = spillTree(head, phase, remainingBytes);
       remainingBytes -= spilled;
     }