修复两个 bug