Correcting a couple of flake8 issues