diff --git a/hw/pci.c b/hw/pci.c
index 2aeaf75a10c6c35436fe7f1a93b7f2a7474f6fbf..9ad63dd94849e1bdda63ddac16369cdde4280e61 100644
--- a/hw/pci.c
+++ b/hw/pci.c
@@ -1290,7 +1290,7 @@ static QObject *pci_get_dev_dict(PCIDevice *dev, PCIBus *bus, int bus_num)
     }
 
     class = pci_get_word(dev->config + PCI_CLASS_DEVICE);
-    if (class == 0x0604) {
+    if (class == PCI_CLASS_BRIDGE_HOST || class == PCI_CLASS_BRIDGE_PCI) {
         QDict *qdict;
         QObject *pci_bridge;