Fixed a flake8 warning