Fixed model class error