py2 compat fix