Small fix in UI