1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/dma/ioat.c b/drivers/dma/ioat.c
index ed83dd9df192..22e9f1911111 100644
--- a/drivers/dma/ioat.c
+++ b/drivers/dma/ioat.c
@@ -98,13 +98,13 @@ static int __devinit ioat_probe(struct pci_dev *pdev,
         if (err)
                 goto err_request_regions;
 
-        err = pci_set_dma_mask(pdev, DMA_64BIT_MASK);
+        err = pci_set_dma_mask(pdev, DMA_BIT_MASK(64));
         if (err)
                 err = pci_set_dma_mask(pdev, DMA_32BIT_MASK);
         if (err)
                 goto err_set_dma_mask;
 
-        err = pci_set_consistent_dma_mask(pdev, DMA_64BIT_MASK);
+        err = pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(64));
         if (err)
                 err = pci_set_consistent_dma_mask(pdev, DMA_32BIT_MASK);
         if (err)