Use correct css class for button