pep8 fix for test failure