diff --git a/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java b/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java
index 7dcaff75..56f4dfa9 100644
--- a/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java
+++ b/src/main/java/com/kamco/cd/kamcoback/scheduler/service/MapSheetInferenceJobService.java
@@ -122,6 +122,7 @@ public class MapSheetInferenceJobService {
         saveInferenceAiDto.setUuid(batchSheet.getUuid());
         saveInferenceAiDto.setStatus(Status.IN_PROGRESS.getId());
         saveInferenceAiDto.setDetectEndCnt(dto.getCompletedJobs().longValue());
+        saveInferenceAiDto.setType(batchSheet.getRunningModelType());
         inferenceResultCoreService.update(saveInferenceAiDto);
       }