diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java b/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java
index 7232e378..24f89f4b 100644
--- a/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java
+++ b/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java
@@ -234,7 +234,9 @@ public class InferenceResultCoreService {
             .orElseThrow(() -> new EntityNotFoundException());
 
     if (request.getType().equals("M1")) {
-      entity.setM1ModelBatchId(request.getBatchId());
+      if (request.getBatchId() != null) {
+        entity.setM1ModelBatchId(request.getBatchId());
+      }
 
       if (request.getModelStartDttm() != null) {
         entity.setM1ModelStartDttm(request.getModelStartDttm());
@@ -245,7 +247,9 @@ public class InferenceResultCoreService {
       }
 
     } else if (request.getType().equals("M2")) {
-      entity.setM2ModelBatchId(request.getBatchId());
+      if (request.getBatchId() != null) {
+        entity.setM2ModelBatchId(request.getBatchId());
+      }
 
       if (request.getModelStartDttm() != null) {
         entity.setM2ModelStartDttm(request.getModelStartDttm());
@@ -256,7 +260,9 @@ public class InferenceResultCoreService {
       }
 
     } else if (request.getType().equals("M3")) {
-      entity.setM3ModelBatchId(request.getBatchId());
+      if (request.getBatchId() != null) {
+        entity.setM3ModelBatchId(request.getBatchId());
+      }
 
       if (request.getModelStartDttm() != null) {
         entity.setM3ModelStartDttm(request.getModelStartDttm());