Some minor UX tweak