1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/iommu/amd_iommu.c b/drivers/iommu/amd_iommu.c
index 72531f008a5e..faf0da4bb3a2 100644
--- a/drivers/iommu/amd_iommu.c
+++ b/drivers/iommu/amd_iommu.c
@@ -248,8 +248,8 @@ static bool check_device(struct device *dev)
         if (!dev || !dev->dma_mask)
                 return false;
 
-        /* No device or no PCI device */
-        if (dev->bus != &pci_bus_type)
+        /* No PCI device */
+        if (!dev_is_pci(dev))
                 return false;
 
         devid = get_device_id(dev);