diff --git a/gluten-substrait/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseTransformStages.scala b/gluten-substrait/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseTransformStages.scala
index 7cf39bf1ae92..2d9b29c252ec 100644
--- a/gluten-substrait/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseTransformStages.scala
+++ b/gluten-substrait/src/main/scala/org/apache/spark/sql/execution/ColumnarCollapseTransformStages.scala
@@ -175,7 +175,8 @@ case class ColumnarCollapseTransformStages(
 
 case class ColumnarInputAdapter(child: SparkPlan)
   extends InputAdapterGenerateTreeStringShim
-  with GlutenPlan {
+  with Convention.KnownBatchType
+  with Convention.KnownRowTypeForSpark33AndLater {
   override def output: Seq[Attribute] = child.output
   override def rowType0(): Convention.RowType = Convention.RowType.None
   override def batchType(): Convention.BatchType =