Correct Model
Correct Model class