fix python2 import error