Fix Check to see if button is pressed