fix bug import error