Slight pep8 fix