py2 fix