Fix a UI glitch