fix class button