fix import error for py2