Fixed incorrect class toggle