Little UI tweak