fix for flake8