Adjust for new button styles
0.1.3