Small ui fixup.