2 files changed, 3 insertions, 3 deletions

diff --git a/virt/kvm/assigned-dev.c b/virt/kvm/assigned-dev.c
index 5819a2708d7e..e05000e200d2 100644
--- a/virt/kvm/assigned-dev.c
+++ b/virt/kvm/assigned-dev.c
@@ -302,7 +302,7 @@ static void kvm_free_assigned_device(struct kvm *kvm,
         else
                 pci_restore_state(assigned_dev->dev);
 
-        assigned_dev->dev->dev_flags &= ~PCI_DEV_FLAGS_ASSIGNED;
+        pci_clear_dev_assigned(assigned_dev->dev);
 
         pci_release_regions(assigned_dev->dev);
         pci_disable_device(assigned_dev->dev);
diff --git a/virt/kvm/iommu.c b/virt/kvm/iommu.c
index 714b94932312..e723bb91aa34 100644
--- a/virt/kvm/iommu.c
+++ b/virt/kvm/iommu.c
@@ -203,7 +203,7 @@ int kvm_assign_device(struct kvm *kvm,
                         goto out_unmap;
         }
 
-        pdev->dev_flags |= PCI_DEV_FLAGS_ASSIGNED;
+        pci_set_dev_assigned(pdev);
 
         dev_info(&pdev->dev, "kvm assign device\n");
 
@@ -229,7 +229,7 @@ int kvm_deassign_device(struct kvm *kvm,
 
         iommu_detach_device(domain, &pdev->dev);
 
-        pdev->dev_flags &= ~PCI_DEV_FLAGS_ASSIGNED;
+        pci_clear_dev_assigned(pdev);
 
         dev_info(&pdev->dev, "kvm deassign device\n");