diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOrcScan.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOrcScan.scala index ef24236b860..7f8b5853c15 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOrcScan.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuOrcScan.scala @@ -840,7 +840,8 @@ class GpuOrcPartitionReader( if (debugDumpPrefix != null) { dumpOrcData(dataBuffer, dataSize) } - val includedColumns = ctx.updatedReadSchema.getFieldNames.asScala + val fieldNames = ctx.updatedReadSchema.getFieldNames.asScala.toArray + val includedColumns = requestedMapping.map(_.map(fieldNames(_))).getOrElse(fieldNames) val parseOpts = ORCOptions.builder() .withTimeUnit(DType.TIMESTAMP_MICROSECONDS) .withNumPyTypes(false)