修复一些 bug