1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/message/i2o/memory.c b/drivers/message/i2o/memory.c
index f5cc95c564e2..9a08d8e45516 100644
--- a/drivers/message/i2o/memory.c
+++ b/drivers/message/i2o/memory.c
@@ -185,7 +185,7 @@ int i2o_dma_alloc(struct device *dev, struct i2o_dma *addr, size_t len)
         int dma_64 = 0;
 
         mutex_lock(&mem_lock);
-        if ((sizeof(dma_addr_t) > 4) && (pdev->dma_mask == DMA_64BIT_MASK)) {
+        if ((sizeof(dma_addr_t) > 4) && (pdev->dma_mask == DMA_BIT_MASK(64))) {
                 dma_64 = 1;
                 if (pci_set_dma_mask(pdev, DMA_32BIT_MASK)) {
                         mutex_unlock(&mem_lock);
@@ -196,7 +196,7 @@ int i2o_dma_alloc(struct device *dev, struct i2o_dma *addr, size_t len)
         addr->virt = dma_alloc_coherent(dev, len, &addr->phys, GFP_KERNEL);
 
         if ((sizeof(dma_addr_t) > 4) && dma_64)
-                if (pci_set_dma_mask(pdev, DMA_64BIT_MASK))
+                if (pci_set_dma_mask(pdev, DMA_BIT_MASK(64)))
                         printk(KERN_WARNING "i2o: unable to set 64-bit DMA");
         mutex_unlock(&mem_lock);