Minor UI tweak