Fix messed up button