diff --git a/bonus-modules/bonus-material/src/main/java/com/bonus/material/basic/domain/BmUnit.java b/bonus-modules/bonus-material/src/main/java/com/bonus/material/basic/domain/BmUnit.java
index 929dd198..414bd429 100644
--- a/bonus-modules/bonus-material/src/main/java/com/bonus/material/basic/domain/BmUnit.java
+++ b/bonus-modules/bonus-material/src/main/java/com/bonus/material/basic/domain/BmUnit.java
@@ -7,6 +7,7 @@ import lombok.ToString;
 import com.bonus.common.core.web.domain.BaseEntity;
 
 import javax.validation.constraints.NotBlank;
+import javax.validation.constraints.NotNull;
 import javax.validation.constraints.Size;
 
 /**
@@ -38,7 +39,7 @@ public class BmUnit extends BaseEntity
 
     /** 单位类型 */
     @ApiModelProperty(value = "单位类型")
-    @NotBlank(message = "单位类型不能为空")
+    @NotNull(message = "单位类型不能为空")
     private Long typeId;
 
     @Excel(name = "单位类型")
@@ -47,7 +48,7 @@ public class BmUnit extends BaseEntity
 
     /** 所属分公司 */
     @ApiModelProperty(value = "所属分公司")
-    @NotBlank(message = "所属分公司不能为空")
+    @NotNull(message = "所属分公司不能为空")
     private Long deptId;
 
     @Excel(name = "所属分公司")