fixed a UI bug