From 22e088b087d02d3ce6d57a2590f8fec380d78f11 Mon Sep 17 00:00:00 2001
From: Alexander Spies <alexander.spies@elastic.co>
Date: Fri, 13 Dec 2024 15:37:45 +0100
Subject: [PATCH] Fix test

---
 muted-tests.yml                                                | 3 ---
 .../xpack/esql/optimizer/LogicalPlanOptimizerTests.java        | 3 +--
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/muted-tests.yml b/muted-tests.yml
index d5b933b96d73b..95beeb7aa8f8d 100644
--- a/muted-tests.yml
+++ b/muted-tests.yml
@@ -317,9 +317,6 @@ tests:
 - class: org.elasticsearch.reservedstate.service.FileSettingsServiceTests
   method: testInvalidJSON
   issue: https://github.com/elastic/elasticsearch/issues/116521
-- class: org.elasticsearch.xpack.esql.optimizer.LogicalPlanOptimizerTests
-  method: testPlanSanityCheckWithBinaryPlans
-  issue: https://github.com/elastic/elasticsearch/issues/118656
 
 # Examples:
 #
diff --git a/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LogicalPlanOptimizerTests.java b/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LogicalPlanOptimizerTests.java
index 7e498eb6654b9..d97a8bb2bc27f 100644
--- a/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LogicalPlanOptimizerTests.java
+++ b/x-pack/plugin/esql/src/test/java/org/elasticsearch/xpack/esql/optimizer/LogicalPlanOptimizerTests.java
@@ -4911,8 +4911,7 @@ public void testPlanSanityCheckWithBinaryPlans() throws Exception {
             """);
 
         var project = as(plan, Project.class);
-        var limit = as(project.child(), Limit.class);
-        var join = as(limit.child(), Join.class);
+        var join = as(project.child(), Join.class);
 
         var joinWithInvalidLeftPlan = join.replaceChildren(join.right(), join.right());
         IllegalStateException e = expectThrows(IllegalStateException.class, () -> logicalOptimizer.optimize(joinWithInvalidLeftPlan));