},
 };
 
-static int stex_set_dma_mask(struct pci_dev * pdev)
-{
-       int ret;
-
-       if (!pci_set_dma_mask(pdev, DMA_BIT_MASK(64))
-               && !pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(64)))
-               return 0;
-       ret = pci_set_dma_mask(pdev, DMA_BIT_MASK(32));
-       if (!ret)
-               ret = pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(32));
-       return ret;
-}
-
 static int stex_request_irq(struct st_hba *hba)
 {
        struct pci_dev *pdev = hba->pdev;
                goto out_release_regions;
        }
 
-       err = stex_set_dma_mask(pdev);
+       err = dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(64));
+       if (err)
+               err = dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(32));
        if (err) {
                printk(KERN_ERR DRV_NAME "(%s): set dma mask failed\n",
                        pci_name(pdev));