diff --git a/core/sap/inc/sap_api.h b/core/sap/inc/sap_api.h
index 14c42f04e11d..c1c5f465c66f 100644
--- a/core/sap/inc/sap_api.h
+++ b/core/sap/inc/sap_api.h
@@ -1048,13 +1048,14 @@ QDF_STATUS wlansap_set_dfs_target_chnl(tHalHandle hHal,
 
 /**
  * wlan_sap_get_phymode() - Returns sap phymode.
- * @ctx:	Pointer to cds Context or Sap Context.
+ * @sap_ctx:	Pointer to Sap Context.
  *
  * This function provides the SAP current phymode.
  *
  * Return: phymode
  */
-eCsrPhyMode wlan_sap_get_phymode(void *ctx);
+eCsrPhyMode wlan_sap_get_phymode(struct sap_context *sap_ctx);
+
 uint32_t wlan_sap_get_vht_ch_width(void *ctx);
 void wlan_sap_set_vht_ch_width(void *ctx, uint32_t vht_channel_width);
 QDF_STATUS wlansap_update_sap_config_add_ie(tsap_Config_t *pConfig,
diff --git a/core/sap/src/sap_module.c b/core/sap/src/sap_module.c
index de78ebb4c927..ba72b96079f2 100644
--- a/core/sap/src/sap_module.c
+++ b/core/sap/src/sap_module.c
@@ -588,10 +588,8 @@ wlansap_set_scan_acs_channel_params(tsap_Config_t *pconfig,
 	return status;
 }
 
-eCsrPhyMode wlan_sap_get_phymode(void *ctx)
+eCsrPhyMode wlan_sap_get_phymode(struct sap_context *sap_ctx)
 {
-	struct sap_context *sap_ctx = CDS_GET_SAP_CB(ctx);
-
 	if (!sap_ctx) {
 		QDF_TRACE(QDF_MODULE_ID_SAP, QDF_TRACE_LEVEL_ERROR,
 			FL("Invalid SAP pointer from ctx"));