modify: 简化代码