Fixed a UI glitch