1 files changed, 4 insertions, 2 deletions

diff --git a/arch/powerpc/platforms/pseries/msi.c b/arch/powerpc/platforms/pseries/msi.c
index 691a154c286d..c8d24f9a6948 100644
--- a/arch/powerpc/platforms/pseries/msi.c
+++ b/arch/powerpc/platforms/pseries/msi.c
@@ -195,6 +195,7 @@ static struct device_node *find_pe_total_msi(struct pci_dev *dev, int *total)
 static struct device_node *find_pe_dn(struct pci_dev *dev, int *total)
 {
         struct device_node *dn;
+        struct pci_dn *pdn;
         struct eeh_dev *edev;
 
         /* Found our PE and assume 8 at that point. */
@@ -204,10 +205,11 @@ static struct device_node *find_pe_dn(struct pci_dev *dev, int *total)
                 return NULL;
 
         /* Get the top level device in the PE */
-        edev = of_node_to_eeh_dev(dn);
+        edev = pdn_to_eeh_dev(PCI_DN(dn));
         if (edev->pe)
                 edev = list_first_entry(&edev->pe->edevs, struct eeh_dev, list);
-        dn = eeh_dev_to_of_node(edev);
+        pdn = eeh_dev_to_pdn(edev);
+        dn = pdn ? pdn->node : NULL;
         if (!dn)
                 return NULL;