Fix pep8 complaint