python2 fix