diff --git a/public/pages/property/unit/unit.html b/public/pages/property/unit/unit.html
index c715828b5..f0753d796 100644
--- a/public/pages/property/unit/unit.html
+++ b/public/pages/property/unit/unit.html
@@ -70,9 +70,8 @@
                     <h5>单元信息</h5>
                     <div class="ibox-tools" style="top:10px;">
                         <button type="button" class="btn btn-primary btn-sm"
-                                v-if="unitInfo.conditions.floorId != null && unitInfo.conditions.floorId != ''"
                                 style="margin-left:10px" v-on:click="openAddUnitModel()">
-                            <i class="fa fa-plus"></i> 添加单元
+                            <i class="fa fa-plus"></i> 添加
                         </button>
                     </div>
                 </div>
diff --git a/public/pages/property/unit/unit.js b/public/pages/property/unit/unit.js
index 721bc7889..b3872c3f3 100644
--- a/public/pages/property/unit/unit.js
+++ b/public/pages/property/unit/unit.js
@@ -71,6 +71,10 @@
                 vc.emit('searchFloor','openSearchFloorModel',{});
             },
             openAddUnitModel:function(){
+                if(!vc.isNotEmpty($that.unitInfo.conditions.floorId)){
+                    vc.toast('请先选择楼栋');
+                    return ;
+                }
                 vc.emit('addUnit','addUnitModel',{
                     floorId:vc.component.unitInfo.conditions.floorId
                 });