From 33acd2d4202ec714cd616068ef1c8d0458242c84 Mon Sep 17 00:00:00 2001
From: Hongze Zhang <hongze.zhang@intel.com>
Date: Mon, 12 Aug 2024 14:00:11 +0800
Subject: [PATCH] fixup

---
 cpp/velox/tests/MemoryManagerTest.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/velox/tests/MemoryManagerTest.cc b/cpp/velox/tests/MemoryManagerTest.cc
index 0af4ba71ff71..bb102dc2d8c3 100644
--- a/cpp/velox/tests/MemoryManagerTest.cc
+++ b/cpp/velox/tests/MemoryManagerTest.cc
@@ -392,7 +392,7 @@ TEST_F(MultiMemoryManagerTest, spill) {
 
   for (auto& vmm : vmms) {
     assertCapacitiesMatch(tmm, vmms);
-    vmm->getMemoryManager()->arbitrator()->shrinkCapacity(allocateSize);
+    vmm->getMemoryManager()->arbitrator()->shrinkCapacity(allocateSize * numAllocations);
     assertCapacitiesMatch(tmm, vmms);
   }