Fixed a flake8 error