diff --git a/core/trino-main/src/main/java/io/trino/sql/planner/EqualityInference.java b/core/trino-main/src/main/java/io/trino/sql/planner/EqualityInference.java
index f67540b07e34..39da9663a0b6 100644
--- a/core/trino-main/src/main/java/io/trino/sql/planner/EqualityInference.java
+++ b/core/trino-main/src/main/java/io/trino/sql/planner/EqualityInference.java
@@ -130,7 +130,7 @@ public EqualityPartition generateEqualitiesPartitionedBy(Set<Symbol> scope)
                         if (scopeRewritten != null) {
                             scopeExpressions.add(scopeRewritten);
                         }
-                        Expression scopeComplementRewritten = rewrite(candidate, expression -> !scope.contains(expression), false);
+                        Expression scopeComplementRewritten = rewrite(candidate, symbol -> !scope.contains(symbol), false);
                         if (scopeComplementRewritten != null) {
                             scopeComplementExpressions.add(scopeComplementRewritten);
                         }