1 files changed, 2 insertions, 2 deletions

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