return 0;
 }
 
+static int blocking_domain_set_dev_pasid(struct iommu_domain *domain,
+                                        struct device *dev, ioasid_t pasid,
+                                        struct iommu_domain *old);
+
 static struct iommu_domain blocking_domain = {
        .type = IOMMU_DOMAIN_BLOCKED,
        .ops = &(const struct iommu_domain_ops) {
                .attach_dev     = blocking_domain_attach_dev,
+               .set_dev_pasid  = blocking_domain_set_dev_pasid,
        }
 };
 
        kfree(dev_pasid);
 }
 
-static void intel_iommu_remove_dev_pasid(struct device *dev, ioasid_t pasid,
-                                        struct iommu_domain *domain)
+static int blocking_domain_set_dev_pasid(struct iommu_domain *domain,
+                                        struct device *dev, ioasid_t pasid,
+                                        struct iommu_domain *old)
 {
        struct device_domain_info *info = dev_iommu_priv_get(dev);
 
        intel_pasid_tear_down_entry(info->iommu, dev, pasid, false);
-       domain_remove_dev_pasid(domain, dev, pasid);
+       domain_remove_dev_pasid(old, dev, pasid);
+
+       return 0;
 }
 
 struct dev_pasid_info *
        .dev_disable_feat       = intel_iommu_dev_disable_feat,
        .is_attach_deferred     = intel_iommu_is_attach_deferred,
        .def_domain_type        = device_def_domain_type,
-       .remove_dev_pasid       = intel_iommu_remove_dev_pasid,
        .pgsize_bitmap          = SZ_4K,
        .page_response          = intel_iommu_page_response,
        .default_domain_ops = &(const struct iommu_domain_ops) {