fix py2 compat