fix python 3.x import error