small bugfix, pep8