Fixed a pep8 issue introduced earlier