Fix classes not being set.