Fixes class addition in the custom toggle demo