修改选取单位的生成代码