PEP8 PEP8 PEP8
Fix tests broken with PEP8