Small button tweak