Fixed the tests for the models.