Fixed a UI bug