UI bugfix