diff --git a/spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala b/spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala index 8a6d876c8..7424f1bfd 100644 --- a/spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala +++ b/spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala @@ -61,7 +61,7 @@ class CometExpressionSuite extends CometTestBase with AdaptiveSparkPlanHelper { readParquetFile(dir.getCanonicalPath) { df => { val decimalLiteral = Decimal(0.00) - val cometDf = df.select($"dec" / decimalLiteral) + val cometDf = df.select($"dec" / decimalLiteral, $"dec" % decimalLiteral) checkSparkAnswerAndOperator(cometDf) } }