diff --git a/wmi_unified_api.c b/wmi_unified_api.c
index 2224df45db..69da2d8a29 100644
--- a/wmi_unified_api.c
+++ b/wmi_unified_api.c
@@ -95,7 +95,7 @@ QDF_STATUS wmi_unified_vdev_delete_send(void *wmi_hdl,
 QDF_STATUS
 wmi_unified_vdev_nss_chain_params_send(void *wmi_hdl,
 				       uint8_t vdev_id,
-				       struct wlan_mlme_nss_chains *user_cfg)
+				       struct vdev_nss_chains *user_cfg)
 {
 	wmi_unified_t wmi_handle = (wmi_unified_t)wmi_hdl;
 
diff --git a/wmi_unified_tlv.c b/wmi_unified_tlv.c
index 60a868a5c0..17ff130c0d 100644
--- a/wmi_unified_tlv.c
+++ b/wmi_unified_tlv.c
@@ -293,7 +293,7 @@ static QDF_STATUS send_vdev_delete_cmd_tlv(wmi_unified_t wmi_handle,
 static QDF_STATUS
 send_vdev_nss_chain_params_cmd_tlv(wmi_unified_t wmi_handle,
 				   uint8_t vdev_id,
-				   struct wlan_mlme_nss_chains *user_cfg)
+				   struct vdev_nss_chains *user_cfg)
 {
 	wmi_vdev_chainmask_config_cmd_fixed_param *cmd;
 	wmi_buf_t buf;