diff --git a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
index b6547e2321..186aab381b 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/MLCommonsRestTestCase.java
@@ -38,7 +38,6 @@
 import java.util.function.Consumer;
 import java.util.stream.Collectors;
 
-import lombok.extern.log4j.Log4j2;
 import org.apache.hc.client5.http.auth.AuthScope;
 import org.apache.hc.client5.http.auth.UsernamePasswordCredentials;
 import org.apache.hc.client5.http.impl.auth.BasicCredentialsProvider;
@@ -104,6 +103,8 @@
 import com.google.gson.Gson;
 import com.google.gson.JsonArray;
 
+import lombok.extern.log4j.Log4j2;
+
 @Log4j2
 public abstract class MLCommonsRestTestCase extends OpenSearchRestTestCase {
     protected Gson gson = new Gson();