flake8 fixes
flake8 fix