diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java
index aaa68da190..8096da1fbc 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/RestBedRockInferenceIT.java
@@ -302,7 +302,9 @@ private void validateOutput(String errorMsg, Map<String, Object> output, String
         List outputList = (List) output.get("output");
         assertEquals(errorMsg, 1, outputList.size());
         assertTrue(errorMsg, outputList.get(0) instanceof Map);
-        String typeErrorMsg = errorMsg + " first element in the output list is type of: " + ((Map<?, ?>)outputList.get(0)).get("data").getClass().getName();
+        String typeErrorMsg = errorMsg
+            + " first element in the output list is type of: "
+            + ((Map<?, ?>) outputList.get(0)).get("data").getClass().getName();
         assertTrue(typeErrorMsg, ((Map<?, ?>) outputList.get(0)).get("data") instanceof List);
         assertEquals(errorMsg, ((Map<?, ?>) outputList.get(0)).get("data_type"), dataType);
     }
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java b/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java
index 58cb1e69ed..239fa9c917 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/RestCohereInferenceIT.java
@@ -88,7 +88,9 @@ private void validateOutput(String errorMsg, Map<String, Object> output, String
         List outputList = (List) output.get("output");
         assertEquals(errorMsg, 2, outputList.size());
         assertTrue(errorMsg, outputList.get(0) instanceof Map);
-        String typeErrorMsg = errorMsg + " first element in the output list is type of: " + ((Map<?, ?>)outputList.get(0)).get("data").getClass().getName();
+        String typeErrorMsg = errorMsg
+            + " first element in the output list is type of: "
+            + ((Map<?, ?>) outputList.get(0)).get("data").getClass().getName();
         assertTrue(typeErrorMsg, ((Map<?, ?>) outputList.get(0)).get("data") instanceof List);
         assertTrue(errorMsg, ((Map<?, ?>) outputList.get(0)).get("data_type").equals(dataType));
     }