diff --git a/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/dal/mysql/dept/DeptMapper.java b/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/dal/mysql/dept/DeptMapper.java
index 595ed075..33d8352b 100644
--- a/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/dal/mysql/dept/DeptMapper.java
+++ b/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/dal/mysql/dept/DeptMapper.java
@@ -63,5 +63,5 @@ public interface DeptMapper extends BaseMapperX<DeptDO> {
     }
 
     List<DeptDO> selectCompany(@Param("flag") String flag,
-                               @Param("type") String type);
+                               @Param("type") List<String> type);
 }
diff --git a/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/dept/DeptServiceImpl.java b/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/dept/DeptServiceImpl.java
index 7dedab46..3bdb1e0f 100644
--- a/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/dept/DeptServiceImpl.java
+++ b/yudao-module-system/yudao-module-system-biz/src/main/java/cn/iocoder/yudao/module/system/service/dept/DeptServiceImpl.java
@@ -373,7 +373,7 @@ public class DeptServiceImpl implements DeptService {
         }
 
         // 根据所在部门信息获取 所在公司信息
-        List<DeptDO> companyDeptList = deptMapper.selectCompany(deptDo.getFlag(), DeptTypeEnum.COMPANY.getValue());
+        List<DeptDO> companyDeptList = deptMapper.selectCompany(deptDo.getFlag(), Arrays.asList(DeptTypeEnum.HEAD_COMPANY.getValue(), DeptTypeEnum.COMPANY.getValue()));
 
         if (CollectionUtil.isEmpty(companyDeptList)) {
             return new DeptDO();
diff --git a/yudao-module-system/yudao-module-system-biz/src/main/resources/mapper/dept/DeptMapper.xml b/yudao-module-system/yudao-module-system-biz/src/main/resources/mapper/dept/DeptMapper.xml
index 4ed0e5c3..b58a08e6 100644
--- a/yudao-module-system/yudao-module-system-biz/src/main/resources/mapper/dept/DeptMapper.xml
+++ b/yudao-module-system/yudao-module-system-biz/src/main/resources/mapper/dept/DeptMapper.xml
@@ -16,7 +16,10 @@
             system_dept
         WHERE
             #{flag} LIKE CONCAT('%', flag, '%')
-            AND type = #{type}
+            AND type IN
+            <foreach collection="type" item="item" open="(" separator="," close=")">
+                #{item}
+            </foreach>
             AND deleted = 0
     </select>
 </mapper>
\ No newline at end of file