UI: Small fix