use the correct class for button