fixed problem with the buttons