model class correction