增加代码，解决Bug