1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/infiniband/hw/nes/nes.c b/drivers/infiniband/hw/nes/nes.c
index ca599767ffbd..7446810446e1 100644
--- a/drivers/infiniband/hw/nes/nes.c
+++ b/drivers/infiniband/hw/nes/nes.c
@@ -478,12 +478,12 @@ static int __devinit nes_probe(struct pci_dev *pcidev, const struct pci_device_i
         }
 
         if ((sizeof(dma_addr_t) > 4)) {
-                ret = pci_set_dma_mask(pcidev, DMA_64BIT_MASK);
+                ret = pci_set_dma_mask(pcidev, DMA_BIT_MASK(64));
                 if (ret < 0) {
                         printk(KERN_ERR PFX "64b DMA mask configuration failed\n");
                         goto bail2;
                 }
-                ret = pci_set_consistent_dma_mask(pcidev, DMA_64BIT_MASK);
+                ret = pci_set_consistent_dma_mask(pcidev, DMA_BIT_MASK(64));
                 if (ret) {
                         printk(KERN_ERR PFX "64b DMA consistent mask configuration failed\n");
                         goto bail2;