diff --git a/core/mac/src/cfg/cfg_api.c b/core/mac/src/cfg/cfg_api.c
index 78db2a88d837..c89eeb00252a 100644
--- a/core/mac/src/cfg/cfg_api.c
+++ b/core/mac/src/cfg/cfg_api.c
@@ -677,7 +677,7 @@ int8_t cfg_get_regulatory_max_transmit_power(tpAniSirGlobal pMac,
  */
 
 QDF_STATUS cfg_get_capability_info(tpAniSirGlobal pMac, uint16_t *pCap,
-				   tpPESession sessionEntry)
+				   struct pe_session *sessionEntry)
 {
 	uint32_t val = 0;
 	tpSirMacCapabilityInfo pCapInfo;
diff --git a/core/mac/src/include/cfg_api.h b/core/mac/src/include/cfg_api.h
index ee5b7aa3f2e2..7e9f463ccce8 100644
--- a/core/mac/src/include/cfg_api.h
+++ b/core/mac/src/include/cfg_api.h
@@ -97,7 +97,7 @@ int8_t cfg_get_regulatory_max_transmit_power(tpAniSirGlobal pMac,
 
 /* / Get capability info */
 QDF_STATUS cfg_get_capability_info(tpAniSirGlobal pMac, uint16_t *pCap,
-				   tpPESession psessionEntry);
+				   struct pe_session *psessionEntry);
 
 /* / Set capability info */
 void cfg_set_capability_info(tpAniSirGlobal, uint16_t);