diff --git a/core/sap/inc/sap_api.h b/core/sap/inc/sap_api.h
index 407a3689c6f8..e5f0ce956528 100644
--- a/core/sap/inc/sap_api.h
+++ b/core/sap/inc/sap_api.h
@@ -1362,7 +1362,15 @@ uint32_t wlansap_get_chan_width(struct sap_context *sap_ctx);
 QDF_STATUS wlansap_set_tx_leakage_threshold(tHalHandle hal,
 			uint16_t tx_leakage_threshold);
 
-QDF_STATUS wlansap_set_invalid_session(void *cds_ctx);
+/*
+ * wlansap_set_invalid_session() - set session ID to invalid
+ * @sap_ctx: pointer to the SAP context
+ *
+ * This function sets session ID to invalid
+ *
+ * Return: QDF_STATUS
+ */
+QDF_STATUS wlansap_set_invalid_session(struct sap_context *sap_ctx);
 
 /**
  * sap_dfs_set_current_channel() - Set current channel params in dfs component
diff --git a/core/sap/src/sap_module.c b/core/sap/src/sap_module.c
index 0b8df8124fec..9f59d6f8c4b8 100644
--- a/core/sap/src/sap_module.c
+++ b/core/sap/src/sap_module.c
@@ -2843,27 +2843,15 @@ QDF_STATUS wlansap_set_tx_leakage_threshold(tHalHandle hal,
 	return QDF_STATUS_SUCCESS;
 }
 
-/*
- * wlansap_set_invalid_session() - set session ID to invalid
- * @cds_ctx: pointer of global context
- *
- * This function sets session ID to invalid
- *
- * Return: QDF_STATUS
- */
-QDF_STATUS
-wlansap_set_invalid_session(void *cds_ctx)
+QDF_STATUS wlansap_set_invalid_session(struct sap_context *sap_ctx)
 {
-	struct sap_context *psapctx;
-
-	psapctx = CDS_GET_SAP_CB(cds_ctx);
-	if (NULL == psapctx) {
+	if (NULL == sap_ctx) {
 		QDF_TRACE(QDF_MODULE_ID_SAP, QDF_TRACE_LEVEL_ERROR,
-			FL("Invalid SAP pointer from pctx"));
+			FL("Invalid SAP pointer"));
 		return QDF_STATUS_E_FAILURE;
 	}
 
-	psapctx->sessionId = CSR_SESSION_ID_INVALID;
+	sap_ctx->sessionId = CSR_SESSION_ID_INVALID;
 
 	return QDF_STATUS_SUCCESS;
 }