Fixed pep8 error in script.