Fix toggle with classes