1 files changed, 3 insertions, 0 deletions

diff --git a/arch/powerpc/platforms/pseries/eeh_cache.c b/arch/powerpc/platforms/pseries/eeh_cache.c
index 9fe2a7de04c..ff6c938f424 100644
--- a/arch/powerpc/platforms/pseries/eeh_cache.c
+++ b/arch/powerpc/platforms/pseries/eeh_cache.c
@@ -307,6 +307,9 @@ void __init pci_addr_cache_build(void)
                 /* Save the BAR's; firmware doesn't restore these after EEH reset */
                 dn = pci_device_to_OF_node(dev);
                 eeh_save_bars(dev, PCI_DN(dn));
+
+                pci_dev_get (dev);  /* matching put is in eeh_remove_device() */
+                PCI_DN(dn)->pcidev = dev;
         }
 
 #ifdef DEBUG