corrected test class ModelTest