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 66fc838..066d7ac 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
@@ -155,7 +155,7 @@ public class ModelTrainMngCoreService {
     } else if (addReq.getModelNo().equals(ModelType.G3.getId())) {
       datasetEntity.setLandCoverCnt(dataset.getSummary().getLandCoverCnt());
     } else if (addReq.getModelNo().equals(ModelType.G4.getId())) {
-      datasetEntity.setSolarCnt(dataset.getSummary().getSolar());
+      datasetEntity.setSolarCnt(dataset.getSummary().getSolarCnt());
     }
 
     datasetEntity.setCreatedUid(userUtil.getId());