Cleanup for flake8.