Fix for pep8 CI failures