UI: Small fix #2