代码更新

修复bug