Fix for new_classes field not being updated