调整代码，修个bug