diff --git a/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java b/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java
index 6825941..fc91be8 100644
--- a/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java
+++ b/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java
@@ -274,14 +274,14 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom {
         .select(
             Projections.constructor(
                 ModelTrainMngDto.ProgressPercent.class,
-                modelTrainJobEntity.id,
+                modelTrainJobEntity.modelId,
                 modelTrainJobEntity.jobType,
                 modelTrainJobEntity.statusCd,
                 totalEpoch.as("totalEpoch"),
                 currentEpoch.as("currentEpoch"),
                 per.as("per")))
         .from(modelTrainJobEntity)
-        .where(modelTrainJobEntity.id.eq(id))
+        .where(modelTrainJobEntity.modelId.eq(id))
         .orderBy(modelTrainJobEntity.attemptNo.desc())
         .fetchFirst();
   }