diff --git a/include/linux/qcom-iommu-util.h b/include/linux/qcom-iommu-util.h
index 291e91f52e21..b90008bfcd13 100644
--- a/include/linux/qcom-iommu-util.h
+++ b/include/linux/qcom-iommu-util.h
@@ -1,6 +1,7 @@
 /* SPDX-License-Identifier: GPL-2.0-only */
 /*
  * Copyright (c) 2020-2021, The Linux Foundation. All rights reserved.
+ * Copyright (c) 2022 Qualcomm Innovation Center, Inc. All rights reserved.
  */
 
 #ifndef __QCOM_IOMMU_UTIL_H
@@ -97,8 +98,9 @@ struct qcom_iommu_ops {
 	int (*enable_s1_translation)(struct iommu_domain *domain);
 	int (*get_mappings_configuration)(struct iommu_domain *domain);
 	struct iommu_ops iommu_ops;
+	struct iommu_domain_ops domain_ops;
 };
-#define to_qcom_iommu_ops(x) (container_of(x, struct qcom_iommu_ops, iommu_ops))
+#define to_qcom_iommu_ops(x) (container_of(x, struct qcom_iommu_ops, domain_ops))
 
 struct device_node *qcom_iommu_group_parse_phandle(struct device *dev);
 int qcom_iommu_generate_dma_regions(struct device *dev,