diff --git a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java
index 43023bae..268694f0 100644
--- a/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java
+++ b/src/main/java/com/kamco/cd/kamcoback/inference/service/InferenceResultService.java
@@ -414,9 +414,9 @@ public class InferenceResultService {
 
     String modelType = "";
 
-    if (modelInfo.getModelType().equals(ModelType.G1.getId())) {
+    if (modelInfo.getModelType().equals(ModelType.M1.getId())) {
       modelType = "G1";
-    } else if (modelInfo.getModelType().equals(ModelType.G2.getId())) {
+    } else if (modelInfo.getModelType().equals(ModelType.M2.getId())) {
       modelType = "G2";
     } else {
       modelType = "G3";
diff --git a/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java b/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java
index 6cf752bc..bbff6f3f 100644
--- a/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java
+++ b/src/main/java/com/kamco/cd/kamcoback/model/dto/ModelMngDto.java
@@ -21,9 +21,9 @@ public class ModelMngDto {
   @Getter
   @AllArgsConstructor
   public enum ModelType implements EnumType {
-    G1("G1"),
-    G2("G2"),
-    G3("G3");
+    M1("M1"),
+    M2("M2"),
+    M3("M3");
 
     private final String desc;