pep8 logic fix