int (*read_config)(struct pci_dn *pdn, int where, int size, u32 *val);
        int (*write_config)(struct pci_dn *pdn, int where, int size, u32 val);
        int (*next_error)(struct eeh_pe **pe);
-       int (*restore_config)(struct pci_dn *pdn);
+       int (*restore_config)(struct eeh_dev *edev);
        int (*notify_resume)(struct pci_dn *pdn);
 };
 
 
 
 static void eeh_restore_dev_state(struct eeh_dev *edev, void *userdata)
 {
-       struct pci_dn *pdn = eeh_dev_to_pdn(edev);
        struct pci_dev *pdev = eeh_dev_to_pci_dev(edev);
        struct pci_dev *dev = userdata;
 
                return;
 
        /* Apply customization from firmware */
-       if (pdn && eeh_ops->restore_config)
-               eeh_ops->restore_config(pdn);
+       if (eeh_ops->restore_config)
+               eeh_ops->restore_config(edev);
 
        /* The caller should restore state for the specified device */
        if (pdev != dev)
 
  */
 static void eeh_restore_one_device_bars(struct eeh_dev *edev, void *flag)
 {
-       struct pci_dn *pdn = eeh_dev_to_pdn(edev);
-
        /* Do special restore for bridges */
        if (edev->mode & EEH_DEV_BRIDGE)
                eeh_restore_bridge_bars(edev);
        else
                eeh_restore_device_bars(edev);
 
-       if (eeh_ops->restore_config && pdn)
-               eeh_ops->restore_config(pdn);
+       if (eeh_ops->restore_config)
+               eeh_ops->restore_config(edev);
 }
 
 /**
 
        return ret;
 }
 
-static int pnv_eeh_restore_config(struct pci_dn *pdn)
+static int pnv_eeh_restore_config(struct eeh_dev *edev)
 {
-       struct eeh_dev *edev = pdn_to_eeh_dev(pdn);
        struct pnv_phb *phb;
        s64 ret = 0;
-       int config_addr = (pdn->busno << 8) | (pdn->devfn);
 
        if (!edev)
                return -EEXIST;
 
        if (ret) {
                pr_warn("%s: Can't reinit PCI dev 0x%x (%lld)\n",
-                       __func__, config_addr, ret);
+                       __func__, edev->bdfn, ret);
                return -EIO;
        }