diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
index a2a23c99e..fe0a253a5 100644
--- a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
+++ b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
@@ -235,7 +235,7 @@ class CometSparkSessionExtensions
           val nativeOp = QueryPlanSerde.operator2Proto(op).get
           CometScanWrapper(nativeOp, op)
 
-        case leaf: LeafExecNode if applyRowToColumnar(leaf) =>
+        case leaf: LeafExecNode if applyRowToColumnar(conf, leaf) =>
           val cometOp = CometRowToColumnarExec(leaf)
           val nativeOp = QueryPlanSerde.operator2Proto(cometOp).get
           CometSinkPlaceHolder(nativeOp, leaf, cometOp)