Correcting flake8 upset