Revert to python2 version