Small PEP8 cleanup