PEP8 issue fixed