cleanup for flake8