修复原代码中的两个bug