PEP8 tweak