Correct flake8 failure