Fix toggle animation