minor fixup of ui