Fix class name toggling