Fix button style getting messed up on click