Small fix in ui