Fixing minor PEP8 issue