修复一下bug