fix bug import