Small pep8 fix