Fix new button style bug