修复部分问题

修复一下代码问题