1 files changed, 5 insertions, 4 deletions

diff --git a/arch/powerpc/platforms/pseries/iommu.c b/arch/powerpc/platforms/pseries/iommu.c
index 7803a19adb31..61d5a17f45c0 100644
--- a/arch/powerpc/platforms/pseries/iommu.c
+++ b/arch/powerpc/platforms/pseries/iommu.c
@@ -49,6 +49,7 @@
 #include <asm/mmzone.h>
 #include <asm/plpar_wrappers.h>
 
+#include "pseries.h"
 
 static void tce_invalidate_pSeries_sw(struct iommu_table *tbl,
                                       __be64 *startp, __be64 *endp)
@@ -1307,16 +1308,16 @@ void iommu_init_early_pSeries(void)
                         ppc_md.tce_free  = tce_free_pSeriesLP;
                 }
                 ppc_md.tce_get   = tce_get_pSeriesLP;
-                ppc_md.pci_dma_bus_setup = pci_dma_bus_setup_pSeriesLP;
-                ppc_md.pci_dma_dev_setup = pci_dma_dev_setup_pSeriesLP;
+                pseries_pci_controller_ops.dma_bus_setup = pci_dma_bus_setup_pSeriesLP;
+                pseries_pci_controller_ops.dma_dev_setup = pci_dma_dev_setup_pSeriesLP;
                 ppc_md.dma_set_mask = dma_set_mask_pSeriesLP;
                 ppc_md.dma_get_required_mask = dma_get_required_mask_pSeriesLP;
         } else {
                 ppc_md.tce_build = tce_build_pSeries;
                 ppc_md.tce_free  = tce_free_pSeries;
                 ppc_md.tce_get   = tce_get_pseries;
-                ppc_md.pci_dma_bus_setup = pci_dma_bus_setup_pSeries;
-                ppc_md.pci_dma_dev_setup = pci_dma_dev_setup_pSeries;
+                pseries_pci_controller_ops.dma_bus_setup = pci_dma_bus_setup_pSeries;
+                pseries_pci_controller_ops.dma_dev_setup = pci_dma_dev_setup_pSeries;
         }