修复几个小的BUG.