Minor UI fix/tweak