1 files changed, 4 insertions, 4 deletions

diff --git a/drivers/uwb/whci.c b/drivers/uwb/whci.c
index 1f8964ed988..2e2784627ad 100644
--- a/drivers/uwb/whci.c
+++ b/drivers/uwb/whci.c
@@ -160,10 +160,10 @@ static int whci_probe(struct pci_dev *pci, const struct pci_device_id *id)
         pci_enable_msi(pci);
         pci_set_master(pci);
         err = -ENXIO;
-        if (!pci_set_dma_mask(pci, DMA_64BIT_MASK))
-                pci_set_consistent_dma_mask(pci, DMA_64BIT_MASK);
-        else if (!pci_set_dma_mask(pci, DMA_32BIT_MASK))
-                pci_set_consistent_dma_mask(pci, DMA_32BIT_MASK);
+        if (!pci_set_dma_mask(pci, DMA_BIT_MASK(64)))
+                pci_set_consistent_dma_mask(pci, DMA_BIT_MASK(64));
+        else if (!pci_set_dma_mask(pci, DMA_BIT_MASK(32)))
+                pci_set_consistent_dma_mask(pci, DMA_BIT_MASK(32));
         else
                 goto error_dma;