修改了一下bug