2 files changed, 3 insertions, 3 deletions

diff --git a/drivers/pci/access.c b/drivers/pci/access.c
index 3af0478c057b..5278ac692cbf 100644
--- a/drivers/pci/access.c
+++ b/drivers/pci/access.c
@@ -472,7 +472,7 @@ EXPORT_SYMBOL_GPL(pci_cfg_access_unlock);
 
 static inline int pcie_cap_version(const struct pci_dev *dev)
 {
-        return dev->pcie_flags_reg & PCI_EXP_FLAGS_VERS;
+        return pcie_caps_reg(dev) & PCI_EXP_FLAGS_VERS;
 }
 
 static inline bool pcie_cap_has_devctl(const struct pci_dev *dev)
@@ -497,7 +497,7 @@ static inline bool pcie_cap_has_sltctl(const struct pci_dev *dev)
         return pcie_cap_version(dev) > 1 ||
                type == PCI_EXP_TYPE_ROOT_PORT ||
                (type == PCI_EXP_TYPE_DOWNSTREAM &&
-                dev->pcie_flags_reg & PCI_EXP_FLAGS_SLOT);
+                pcie_caps_reg(dev) & PCI_EXP_FLAGS_SLOT);
 }
 
 static inline bool pcie_cap_has_rtctl(const struct pci_dev *dev)
diff --git a/drivers/pci/pcie/portdrv_core.c b/drivers/pci/pcie/portdrv_core.c
index b42133afca98..31063ac30992 100644
--- a/drivers/pci/pcie/portdrv_core.c
+++ b/drivers/pci/pcie/portdrv_core.c
@@ -272,7 +272,7 @@ static int get_port_device_capability(struct pci_dev *dev)
 
         /* Hot-Plug Capable */
         if ((cap_mask & PCIE_PORT_SERVICE_HP) &&
-            dev->pcie_flags_reg & PCI_EXP_FLAGS_SLOT) {
+            pcie_caps_reg(dev) & PCI_EXP_FLAGS_SLOT) {
                 pcie_capability_read_dword(dev, PCI_EXP_SLTCAP, &reg32);
                 if (reg32 & PCI_EXP_SLTCAP_HPC) {
                         services |= PCIE_PORT_SERVICE_HP;