diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
index ddfee9f..b327132 100644
--- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
@@ -112,6 +112,7 @@ public class ModelTrainMngCoreService {
       entity.setStep1State(TrainStatusType.IN_PROGRESS.getId());
     } else {
       entity.setStatusCd(TrainStatusType.READY.getId());
+      entity.setStep1State(TrainStatusType.READY.getId());
     }
 
     entity.setCreatedUid(userUtil.getId());