Fixed failing flake8