Fixed flake8 compatibility