Fix model classes.