diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala index 1e868401e..6199f474e 100644 --- a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala +++ b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala @@ -225,7 +225,8 @@ class CometSparkSessionExtensions private def transform(plan: SparkPlan): SparkPlan = { def transform1(op: SparkPlan): Option[Operator] = { if (op.children.forall(_.isInstanceOf[CometNativeExec])) { - QueryPlanSerde.operator2Proto(op, + QueryPlanSerde.operator2Proto( + op, op.children.map(_.asInstanceOf[CometNativeExec].nativeOp): _*) } else { None