Fixed typo in test_model.py from last commit.