Python 2 compat fix