diff --git a/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanLambdaFunctionsTranslatorTestSuite.scala b/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanLambdaFunctionsTranslatorTestSuite.scala
index 5b34a662b..9c3c1c8a0 100644
--- a/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanLambdaFunctionsTranslatorTestSuite.scala
+++ b/ppl-spark-integration/src/test/scala/org/opensearch/flint/spark/ppl/PPLLogicalPlanLambdaFunctionsTranslatorTestSuite.scala
@@ -181,7 +181,7 @@ class PPLLogicalPlanLambdaFunctionsTranslatorTestSuite
       planTransformer.visit(
         plan(
           pplParser,
-          """source=t | eval a = json_array(1, 2, 3), b = transform(a, 0, (x, y) -> x + y, x -> x * 10)""".stripMargin),
+          """source=t | eval a = json_array(1, 2, 3), b = reduce(a, 0, (x, y) -> x + y, x -> x * 10)""".stripMargin),
         context)
     val table = UnresolvedRelation(Seq("t"))
     val jsonFunc =
@@ -199,7 +199,7 @@ class PPLLogicalPlanLambdaFunctionsTranslatorTestSuite
     val aliasB =
       Alias(
         UnresolvedFunction(
-          "transform",
+          "reduce",
           Seq(UnresolvedAttribute("a"), Literal(0), mergeLambda, finishLambda),
           false),
         "b")()