diff --git a/csrc/velox/column.cpp b/csrc/velox/column.cpp
index 3a852b59c..2ad88f7a8 100644
--- a/csrc/velox/column.cpp
+++ b/csrc/velox/column.cpp
@@ -553,9 +553,9 @@ velox::core::QueryCtx& TorchArrowGlobalStatic::queryContext() {
 }
 
 velox::core::ExecCtx& TorchArrowGlobalStatic::execContext() {
-  static auto pool = velox::memory::getDefaultScopedMemoryPool();
   static velox::core::ExecCtx execContext(
-      pool.get(), &TorchArrowGlobalStatic::queryContext());
+      TorchArrowGlobalStatic::rootMemoryPool(),
+      &TorchArrowGlobalStatic::queryContext());
   return execContext;
 }