flake8 fixes.