fix get Model class