minor pep8 improvement