diff --git a/modules/i3plus-pojo-model/src/main/java/cn/estsh/i3plus/pojo/model/wms/WmsMessageStyleModel.java b/modules/i3plus-pojo-model/src/main/java/cn/estsh/i3plus/pojo/model/wms/WmsMessageStyleModel.java
index 1911b04..ee8cdc6 100644
--- a/modules/i3plus-pojo-model/src/main/java/cn/estsh/i3plus/pojo/model/wms/WmsMessageStyleModel.java
+++ b/modules/i3plus-pojo-model/src/main/java/cn/estsh/i3plus/pojo/model/wms/WmsMessageStyleModel.java
@@ -2,7 +2,6 @@ package cn.estsh.i3plus.pojo.model.wms;
 
 import cn.estsh.i3plus.pojo.base.enumutil.WmsEnumUtil;
 import io.swagger.annotations.ApiParam;
-import io.swagger.models.auth.In;
 import lombok.Data;
 
 import java.io.Serializable;
@@ -39,15 +38,12 @@ public class WmsMessageStyleModel implements Serializable {
 
     public WmsMessageStyleModel() {
     }
+
     public WmsMessageStyleModel(String message, Integer selectQty) {
         this.message = message;
-        if(selectQty==1){
-        this.wsStatus=10;
-     }
-     if(selectQty==2){
-         this.wsStatus=20;
-     }
+        this.wsStatus = selectQty;
     }
+
     public WmsMessageStyleModel(String message) {
         this.message = message;
     }