diff --git a/pineapple-kiwi-cnss.dtsi b/pineapple-kiwi-cnss.dtsi
index 33e0ac1f..1eebf296 100644
--- a/pineapple-kiwi-cnss.dtsi
+++ b/pineapple-kiwi-cnss.dtsi
@@ -172,20 +172,23 @@
 };
 
 &pcie0_rp {
-	#address-cells = <5>;
-	#size-cells = <0>;
 
 	cnss_pci0: cnss_pci0 {
 		reg = <0 0 0 0 0>;
 		qcom,iommu-group = <&cnss_pci_iommu_group0>;
-		memory-region = <&cnss_wlan_mem>;
+		memory-region = <&cnss_wlan_mem &cnss_pci0_iommu_region_partition>;
 
 		#address-cells = <1>;
 		#size-cells = <1>;
 
+		cnss_pci0_iommu_region_partition: cnss_pci0_iommu_region_partition {
+			/* address-cells =3 size-cells=2 from pineapple-pcie.dtsi */
+			iommu-addresses = <&cnss_pci0 0x0 0x0 0x0 0x0 0x98000000>,
+					  <&cnss_pci0 0x0 0x0 0xB0000000 0x0 0x50000000>;
+		};
+
 		cnss_pci_iommu_group0: cnss_pci_iommu_group0 {
 			qcom,iommu-msi-size = <0x1000>;
-			qcom,iommu-dma-addr-pool = <0x98000000 0x18000000>;
 			qcom,iommu-geometry = <0x98000000 0x18010000>;
 			qcom,iommu-dma = "fastmap";
 			qcom,iommu-pagetable = "coherent";