Buttons fixes