diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_embedding/TextEmbeddingDenseModelTest.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_embedding/TextEmbeddingDenseModelTest.java
index 07a9bc2e09..8fd3892bea 100644
--- a/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_embedding/TextEmbeddingDenseModelTest.java
+++ b/ml-algorithms/src/test/java/org/opensearch/ml/engine/algorithms/text_embedding/TextEmbeddingDenseModelTest.java
@@ -348,8 +348,11 @@ public void predict_BeforeInitingModel() {
     public void test_async_inference() {
         exceptionRule.expect(IllegalStateException.class);
         exceptionRule.expectMessage("Method is not implemented");
-        textEmbeddingDenseModel.asyncPredict(MLInput.builder().algorithm(FunctionName.TEXT_EMBEDDING).inputDataset(inputDataSet).build(), mock(
-            ActionListener.class));
+        textEmbeddingDenseModel
+            .asyncPredict(
+                MLInput.builder().algorithm(FunctionName.TEXT_EMBEDDING).inputDataset(inputDataSet).build(),
+                mock(ActionListener.class)
+            );
     }
 
     @After