flake8 fix