修复code的bug