PEP8
Fix for test