Fixed the models_test