diff --git a/ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java b/ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java
index 2e4b2b9927..e5506e6c12 100644
--- a/ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java
+++ b/ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java
@@ -610,21 +610,21 @@ public void testParseCommand() {
   }
 
   @Test
-  public void testStringLiteralHighlightCommand() {
+  public void testQuotedHighlightCommand() {
     assertEqual("source=t | highlight('FieldA')",
         new Highlight(
             alias("highlight('FieldA')",
-                highlight(stringLiteral("FieldA"))))
+                highlight(stringLiteral("'FieldA'"))))
             .attach(relation("t"))
     );
   }
 
   @Test
-  public void testQualifiedNameHighlightCommand() {
+  public void testUnquotedHighlightCommand() {
     assertEqual("source=t | highlight(FieldA)",
         new Highlight(
             alias("highlight(FieldA)",
-            highlight(qualifiedName("FieldA"))))
+            highlight(stringLiteral("FieldA"))))
                 .attach(relation("t"))
     );
   }