small ui fixup