Small bugfix in ui.