diff --git a/core/sap/inc/sap_api.h b/core/sap/inc/sap_api.h
index 30e8dbcc5c..94ba27ef96 100644
--- a/core/sap/inc/sap_api.h
+++ b/core/sap/inc/sap_api.h
@@ -959,8 +959,18 @@ QDF_STATUS wlan_sap_update_next_channel(struct sap_context *sap_ctx,
 QDF_STATUS wlan_sap_set_pre_cac_status(struct sap_context *sap_ctx,
 				       bool status, tHalHandle handle);
 
-QDF_STATUS wlan_sap_set_chan_before_pre_cac(void *ctx,
-		uint8_t chan_before_pre_cac);
+/**
+ * wlan_sap_set_chan_before_pre_cac() - Save the channel before pre cac
+ * @sap_ctx: SAP context
+ * @chan_before_pre_cac: Channel before pre cac
+ *
+ * Saves the channel that was in use before pre cac operation
+ *
+ * Return: QDF_STATUS
+ */
+QDF_STATUS wlan_sap_set_chan_before_pre_cac(struct sap_context *sap_ctx,
+					    uint8_t chan_before_pre_cac);
+
 QDF_STATUS wlan_sap_set_pre_cac_complete_status(void *ctx, bool status);
 bool wlan_sap_is_pre_cac_active(tHalHandle handle);
 QDF_STATUS wlan_sap_get_pre_cac_vdev_id(tHalHandle handle, uint8_t *vdev_id);
diff --git a/core/sap/src/sap_module.c b/core/sap/src/sap_module.c
index 9bdec2cd98..e19eb6bb1a 100644
--- a/core/sap/src/sap_module.c
+++ b/core/sap/src/sap_module.c
@@ -1631,20 +1631,9 @@ QDF_STATUS wlan_sap_set_pre_cac_status(struct sap_context *sap_ctx,
 	return QDF_STATUS_SUCCESS;
 }
 
-/**
- * wlan_sap_set_chan_before_pre_cac() - Save the channel before pre cac
- * @ctx: SAP context
- * @chan_before_pre_cac: Channel before pre cac
- *
- * Saves the channel that was in use before pre cac operation
- *
- * Return: QDF_STATUS
- */
-QDF_STATUS wlan_sap_set_chan_before_pre_cac(void *ctx,
-					uint8_t chan_before_pre_cac)
+QDF_STATUS wlan_sap_set_chan_before_pre_cac(struct sap_context *sap_ctx,
+					    uint8_t chan_before_pre_cac)
 {
-	struct sap_context *sap_ctx = CDS_GET_SAP_CB(ctx);
-
 	if (!sap_ctx) {
 		QDF_TRACE(QDF_MODULE_ID_SAP, QDF_TRACE_LEVEL_ERROR,
 			  "%s: Invalid SAP pointer", __func__);