Fix PEP compliance