fix python2 support
tmp