Fix import code.