BLD: fixed import error