Fixed a UI bug

Critical fix