Fix tests PEP-8