Fix import logic.