fix model class