UI bug 修复