Fixed a couple flake8 warnings.