Update models_tests.py