Fix pep8 warning