Fixed issue with classes buttons