Fix flake8 error in test_model51.py