Fix py2 compat bug