修复了上面的bug