PEP8 fix - not sure if this is correct