diff --git a/core/trino-main/src/test/java/io/trino/sql/planner/assertions/PlanMatchPattern.java b/core/trino-main/src/test/java/io/trino/sql/planner/assertions/PlanMatchPattern.java
index ebb54653c1f9..4919c4fef62a 100644
--- a/core/trino-main/src/test/java/io/trino/sql/planner/assertions/PlanMatchPattern.java
+++ b/core/trino-main/src/test/java/io/trino/sql/planner/assertions/PlanMatchPattern.java
@@ -744,7 +744,7 @@ public static SymbolAlias symbol(String alias)
         return new SymbolAlias(alias);
     }
 
-    public static PlanMatchPattern filter(String expectedPredicate, PlanMatchPattern source)
+    public static PlanMatchPattern filter(@Language("SQL") String expectedPredicate, PlanMatchPattern source)
     {
         return filter(PlanBuilder.expression(expectedPredicate), source);
     }
diff --git a/core/trino-main/src/test/java/io/trino/sql/planner/iterative/rule/test/PlanBuilder.java b/core/trino-main/src/test/java/io/trino/sql/planner/iterative/rule/test/PlanBuilder.java
index 2dcd30ed4b67..86a28d8d8766 100644
--- a/core/trino-main/src/test/java/io/trino/sql/planner/iterative/rule/test/PlanBuilder.java
+++ b/core/trino-main/src/test/java/io/trino/sql/planner/iterative/rule/test/PlanBuilder.java
@@ -114,6 +114,7 @@
 import io.trino.testing.TestingMetadata.TestingTableHandle;
 import io.trino.testing.TestingTableExecuteHandle;
 import io.trino.testing.TestingTransactionHandle;
+import org.intellij.lang.annotations.Language;
 
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -1415,12 +1416,12 @@ public RemoteSourceNode remoteSource(
         return new RemoteSourceNode(idAllocator.getNextId(), sourceFragmentIds, outputs, orderingScheme, exchangeType, retryPolicy);
     }
 
-    public static Expression expression(String sql)
+    public static Expression expression(@Language("SQL") String sql)
     {
         return ExpressionUtils.rewriteIdentifiersToSymbolReferences(new SqlParser().createExpression(sql, new ParsingOptions()));
     }
 
-    public static List<Expression> expressions(String... expressions)
+    public static List<Expression> expressions(@Language("SQL") String... expressions)
     {
         return Stream.of(expressions)
                 .map(PlanBuilder::expression)