改善一个小BUG