diff --git a/common/src/test/java/org/opensearch/ml/common/output/execute/anomalylocalization/AnomalyLocalizationOutputTests.java b/common/src/test/java/org/opensearch/ml/common/output/execute/anomalylocalization/AnomalyLocalizationOutputTests.java
index 5be23ca9dc..72fe1bfd80 100644
--- a/common/src/test/java/org/opensearch/ml/common/output/execute/anomalylocalization/AnomalyLocalizationOutputTests.java
+++ b/common/src/test/java/org/opensearch/ml/common/output/execute/anomalylocalization/AnomalyLocalizationOutputTests.java
@@ -47,9 +47,7 @@ public void testWriteable() throws Exception {
     @Test
     public void testXContent() throws Exception {
         XContentBuilder builder = XContentFactory.jsonBuilder();
-        builder.startObject();
         builder = output.toXContent(builder, null);
-        builder.endObject();
         String json = builder.toString();
         XContentParser parser = XContentType.JSON.xContent().createParser(NamedXContentRegistry.EMPTY, null, json);
         AnomalyLocalizationOutput newOutput = AnomalyLocalizationOutput.parse(parser);
diff --git a/common/src/test/java/org/opensearch/ml/common/output/execute/samplecalculator/LocalSampleCalculatorOutputTest.java b/common/src/test/java/org/opensearch/ml/common/output/execute/samplecalculator/LocalSampleCalculatorOutputTest.java
index ec9e2a43e4..4215bd9fbd 100644
--- a/common/src/test/java/org/opensearch/ml/common/output/execute/samplecalculator/LocalSampleCalculatorOutputTest.java
+++ b/common/src/test/java/org/opensearch/ml/common/output/execute/samplecalculator/LocalSampleCalculatorOutputTest.java
@@ -30,9 +30,7 @@ public void setUp() {
     @Test
     public void toXContent() throws IOException {
         XContentBuilder builder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
-        builder.startObject();
         output.toXContent(builder, ToXContent.EMPTY_PARAMS);
-        builder.endObject();
         String jsonStr = builder.toString();
         assertEquals("{\"result\":1.0}", jsonStr);
     }
@@ -41,9 +39,7 @@ public void toXContent() throws IOException {
     public void toXContent_EmptyOutput() throws IOException {
         LocalSampleCalculatorOutput output = LocalSampleCalculatorOutput.builder().build();
         XContentBuilder builder = MediaTypeRegistry.contentBuilder(XContentType.JSON);
-        builder.startObject();
         output.toXContent(builder, ToXContent.EMPTY_PARAMS);
-        builder.endObject();
         String jsonStr = builder.toString();
         assertEquals("{}", jsonStr);
     }
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestMLExecuteActionTests.java b/plugin/src/test/java/org/opensearch/ml/rest/RestMLExecuteActionTests.java
index 456c46bfb2..86f177a665 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/RestMLExecuteActionTests.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/RestMLExecuteActionTests.java
@@ -21,7 +21,11 @@
 import static org.opensearch.ml.utils.TestHelper.getMetricsCorrelationRestRequest;
 
 import java.io.IOException;
-import java.util.*;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 import org.junit.Before;
 import org.mockito.ArgumentCaptor;