1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/xen/xen-pciback/pci_stub.c b/drivers/xen/xen-pciback/pci_stub.c
index 259ba2661543..017069a455d4 100644
--- a/drivers/xen/xen-pciback/pci_stub.c
+++ b/drivers/xen/xen-pciback/pci_stub.c
@@ -133,7 +133,7 @@ static void pcistub_device_release(struct kref *kref)
         xen_pcibk_config_free_dyn_fields(dev);
         xen_pcibk_config_free_dev(dev);
 
-        dev->dev_flags &= ~PCI_DEV_FLAGS_ASSIGNED;
+        pci_clear_dev_assigned(dev);
         pci_dev_put(dev);
 
         kfree(psdev);
@@ -413,7 +413,7 @@ static int pcistub_init_device(struct pci_dev *dev)
         dev_dbg(&dev->dev, "reset device\n");
         xen_pcibk_reset_device(dev);
 
-        dev->dev_flags |= PCI_DEV_FLAGS_ASSIGNED;
+        pci_set_dev_assigned(dev);
         return 0;
 
 config_release: