diff --git a/wmi_unified_api.h b/wmi_unified_api.h
index f200390a2d..eacaee6097 100644
--- a/wmi_unified_api.h
+++ b/wmi_unified_api.h
@@ -1107,7 +1107,8 @@ QDF_STATUS wmi_unified_init_cmd_send(void *wmi_hdl,
 
 bool wmi_service_enabled(void *wmi_hdl, uint32_t service_id);
 
-QDF_STATUS wmi_save_service_bitmap(void *wmi_hdl, void *evt_buf);
+QDF_STATUS wmi_save_service_bitmap(void *wmi_hdl, void *evt_buf,
+				   void *bitmap_buf);
 
 QDF_STATUS wmi_save_fw_version(void *wmi_hdl, void *evt_buf);
 
diff --git a/wmi_unified_priv.h b/wmi_unified_priv.h
index 58d835e4b2..fa6e59676f 100644
--- a/wmi_unified_priv.h
+++ b/wmi_unified_priv.h
@@ -904,7 +904,7 @@ QDF_STATUS (*send_lteu_config_cmd)(wmi_unified_t wmi_handle,
 QDF_STATUS (*send_set_ps_mode_cmd)(wmi_unified_t wmi_handle,
 		       struct set_ps_mode_params *param);
 void (*save_service_bitmap)(wmi_unified_t wmi_handle,
-		void *evt_buf);
+		void *evt_buf,  void *bitmap_buf);
 bool (*is_service_enabled)(wmi_unified_t wmi_handle,
 	uint32_t service_id);
 QDF_STATUS (*get_target_cap_from_service_ready)(wmi_unified_t wmi_handle,