A bit of flake8 clean up.