Fix PEP8 in test