Minor UI bugfix