fixing the buttons