diff --git a/core/hdd/src/wlan_hdd_driver_ops.c b/core/hdd/src/wlan_hdd_driver_ops.c
index d1a4a2ea7589..7c431396114e 100644
--- a/core/hdd/src/wlan_hdd_driver_ops.c
+++ b/core/hdd/src/wlan_hdd_driver_ops.c
@@ -322,6 +322,7 @@ static void hdd_init_qdf_ctx(struct device *dev, void *bdev,
 	qdf_dev->drv_hdl = bdev;
 	qdf_dev->bus_type = bus_type;
 	qdf_dev->bid = bid;
+	cds_smmu_mem_map_setup(qdf_dev);
 }
 
 /**
@@ -390,7 +391,6 @@ static int wlan_hdd_probe(struct device *dev, void *bdev,
 		cds_set_load_in_progress(true);
 
 	hdd_init_qdf_ctx(dev, bdev, bus_type, (const struct hif_bus_id *)bid);
-	cds_smmu_mem_map_setup(cds_get_context(QDF_MODULE_ID_QDF_DEVICE));
 
 	if (reinit) {
 		ret = hdd_wlan_re_init();