Mybe fix a small pep8 issue