Fix test

Signed-off-by: Andy Kwok <[email protected]>
opensearch-project · Jan 10, 2025 · 302ed85 · 302ed85
1 parent b7acfe1
commit 302ed85
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/...n/java/org/opensearch/sql/opensearch/executor/protector/OpenSearchExecutionProtector.java b/...n/java/org/opensearch/sql/opensearch/executor/protector/OpenSearchExecutionProtector.java
@@ -97,12 +97,14 @@ public PhysicalPlan visitRemove(RemoveOperator node, Object context) {
 
   @Override
   public PhysicalPlan visitEval(EvalOperator node, Object context) {
-    OpenSearchEvalOperator evalOperator = (OpenSearchEvalOperator) node;
-    return doProtect(
-        new OpenSearchEvalOperator(
-            visitInput(evalOperator.getInput(), context),
-            evalOperator.getExpressionList(),
-            evalOperator.getNodeClient()));
+    if (node instanceof OpenSearchEvalOperator evalOperator) {
+      return doProtect(
+              new OpenSearchEvalOperator(
+                      visitInput(evalOperator.getInput(), context),
+                      evalOperator.getExpressionList(),
+                      evalOperator.getNodeClient()));
+    }
+    return new EvalOperator(visitInput(node.getInput(), context), node.getExpressionList());
   }
 
   @Override