Update for PR comments
Fix PEP error