py2 fixes