PEP8 code fix