Fixed models_test.py