Python: small fix