Fixes problems with flake8 due to pep8 update