Minor fix PEP8