diff --git a/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala b/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala index f664c8a1d696..ba24786c2f25 100644 --- a/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala +++ b/gluten-substrait/src/main/scala/org/apache/gluten/execution/TakeOrderedAndProjectExecTransformer.scala @@ -143,6 +143,7 @@ case class TakeOrderedAndProjectExecTransformer( } else { val limitStagePlan = WholeStageTransformer(limitBeforeShuffle)(transformStageCounter.incrementAndGet()) + limitStagePlan.copyTagsFrom(child) val shuffleExec = ShuffleExchangeExec(SinglePartition, limitStagePlan) val transformedShuffleExec = ColumnarShuffleExchangeExec(shuffleExec, limitStagePlan, shuffleExec.child.output)