PEP8 warning fixed