fix custom button class issue