修复之前的bug