UX bugfix