diff --git a/common/src/test/java/org/opensearch/ml/common/transport/model_group/MLModelGroupGetResponseTest.java b/common/src/test/java/org/opensearch/ml/common/transport/model_group/MLModelGroupGetResponseTest.java
index 2f2f01f976..0143c1851e 100644
--- a/common/src/test/java/org/opensearch/ml/common/transport/model_group/MLModelGroupGetResponseTest.java
+++ b/common/src/test/java/org/opensearch/ml/common/transport/model_group/MLModelGroupGetResponseTest.java
@@ -66,7 +66,7 @@ public void toXContentTest() throws IOException {
     }
 
     @Test
-    public void fromActionResponseWithMLModelGetResponse_Success() {
+    public void fromActionResponseWithMLModelGroupGetResponse_Success() {
         MLModelGroupGetResponse mlModelGroupGetResponse = MLModelGroupGetResponse.builder().mlModelGroup(mlModelGroup).build();
         MLModelGroupGetResponse mlModelGroupGetResponseFromActionResponse = MLModelGroupGetResponse.fromActionResponse(mlModelGroupGetResponse);
         assertSame(mlModelGroupGetResponse, mlModelGroupGetResponseFromActionResponse);
diff --git a/plugin/src/test/java/org/opensearch/ml/rest/MLModelGroupRestIT.java b/plugin/src/test/java/org/opensearch/ml/rest/MLModelGroupRestIT.java
index 4f04fe92cf..ac8b8057a5 100644
--- a/plugin/src/test/java/org/opensearch/ml/rest/MLModelGroupRestIT.java
+++ b/plugin/src/test/java/org/opensearch/ml/rest/MLModelGroupRestIT.java
@@ -7,8 +7,11 @@
 
 package org.opensearch.ml.rest;
 
-import com.google.common.base.Throwables;
-import com.google.common.collect.ImmutableList;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Map;
+
 import org.apache.hc.core5.http.HttpHeaders;
 import org.apache.hc.core5.http.HttpHost;
 import org.apache.hc.core5.http.message.BasicHeader;
@@ -25,10 +28,8 @@
 import org.opensearch.ml.common.transport.model_group.MLUpdateModelGroupInput;
 import org.opensearch.ml.utils.TestHelper;
 
-import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Map;
+import com.google.common.base.Throwables;
+import com.google.common.collect.ImmutableList;
 
 public class MLModelGroupRestIT extends MLCommonsRestTestCase {