diff --git a/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala b/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala
index 36ce0ffae135..114de2b623fd 100644
--- a/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala
+++ b/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala
@@ -150,6 +150,7 @@ class IteratorApiImpl extends IteratorApi with Logging {
 
     Iterators
       .wrap(resIter.asScala)
+      .protectInvocationFlow()
       .recycleIterator {
         updateNativeMetrics(resIter.getMetrics)
         updateInputMetrics(TaskContext.get().taskMetrics().inputMetrics)