Small tweak to buttons