diff --git a/drivers/soc/qcom/icnss.c b/drivers/soc/qcom/icnss.c
index 37204dcbc065..7067c5733773 100644
--- a/drivers/soc/qcom/icnss.c
+++ b/drivers/soc/qcom/icnss.c
@@ -3557,18 +3557,16 @@ unsigned int icnss_socinfo_get_serial_number(struct device *dev)
 }
 EXPORT_SYMBOL(icnss_socinfo_get_serial_number);
 
-int icnss_set_wlan_mac_address(struct device *dev,
-			       const u8 *in, uint32_t len)
+int icnss_set_wlan_mac_address(const u8 *in, const uint32_t len)
 {
-	struct icnss_priv *priv = dev_get_drvdata(dev);
+	struct icnss_priv *priv = penv;
 	uint32_t no_of_mac_addr;
 	struct icnss_wlan_mac_addr *addr = NULL;
 	int iter;
 	u8 *temp = NULL;
 
-	if (priv->magic != ICNSS_MAGIC) {
-		icnss_pr_err("Invalid drvdata: dev %p, data %p, magic 0x%x\n",
-			     dev, priv, priv->magic);
+	if (!priv) {
+		icnss_pr_err("Priv data is NULL\n");
 		return -EINVAL;
 	}
 
diff --git a/include/soc/qcom/icnss.h b/include/soc/qcom/icnss.h
index 29990f036552..9c38b9aa5627 100644
--- a/include/soc/qcom/icnss.h
+++ b/include/soc/qcom/icnss.h
@@ -124,8 +124,7 @@ extern int icnss_get_wlan_unsafe_channel(u16 *unsafe_ch_list, u16 *ch_count,
 extern int icnss_wlan_set_dfs_nol(const void *info, u16 info_len);
 extern int icnss_wlan_get_dfs_nol(void *info, u16 info_len);
 extern bool icnss_is_qmi_disable(void);
-extern int icnss_set_wlan_mac_address(struct device *dev,
-				      const u8 *in, uint32_t len);
+extern int icnss_set_wlan_mac_address(const u8 *in, const uint32_t len);
 extern u8 *icnss_get_wlan_mac_address(struct device *dev, uint32_t *num);
 
 #endif /* _ICNSS_WLAN_H_ */