Minor fix - pep8