Small fix in UI.