diff --git a/src/main/java/com/kamco/cd/training/model/service/ModelTrainDetailService.java b/src/main/java/com/kamco/cd/training/model/service/ModelTrainDetailService.java
index 5d57b60..6e53452 100644
--- a/src/main/java/com/kamco/cd/training/model/service/ModelTrainDetailService.java
+++ b/src/main/java/com/kamco/cd/training/model/service/ModelTrainDetailService.java
@@ -81,6 +81,7 @@ public class ModelTrainDetailService {
       datasetIds.add(mappingDataset.getDatasetId());
     }
     datasetReq.setIds(datasetIds);
+    datasetReq.setModelNo(modelInfo.getModelNo());
 
     if (modelInfo.getModelNo().equals("G1")) {
       dataSets = mngCoreService.getDatasetSelectG1List(datasetReq);