diff --git a/hif/src/ce/ce_assignment.h b/hif/src/ce/ce_assignment.h
index 9f9f3db029..ecf74d05bd 100644
--- a/hif/src/ce/ce_assignment.h
+++ b/hif/src/ce/ce_assignment.h
@@ -1580,7 +1580,7 @@ static struct CE_attr host_ce_config_wlan_qcn9224[] = {
 	/* Target CV prefetch */
 	{/*CE13*/ (CE_ATTR_FLAGS | CE_ATTR_DISABLE_INTR), 0, 0, 0, 0, NULL,},
 	/* target->host WMI logging, Direc-DMA */
-	{/*CE14*/ (CE_ATTR_FLAGS), 0, 0, 0, 0, NULL,},
+	{/*CE14*/ (CE_ATTR_FLAGS), 0, 0, 2048, 128, NULL,},
 	/* Reserved for customer use */
 	{/*CE15*/ (CE_ATTR_FLAGS | CE_ATTR_DISABLE_INTR), 0, 0, 0, 0, NULL,},
 };
diff --git a/wmi/src/wmi_unified.c b/wmi/src/wmi_unified.c
index eaba07c092..bc53e3524f 100644
--- a/wmi/src/wmi_unified.c
+++ b/wmi/src/wmi_unified.c
@@ -2627,7 +2627,8 @@ static void wmi_control_rx(void *ctx, HTC_PACKET *htc_packet)
 	wmi_process_control_rx(wmi_handle, evt_buf);
 }
 
-#if defined(WLAN_FEATURE_WMI_DIAG_OVER_CE7)
+#if defined(WLAN_FEATURE_WMI_DIAG_OVER_CE7) || \
+	defined(WLAN_DIAG_AND_DBR_OVER_SEPARATE_CE)
 /**
  * wmi_control_diag_rx() - process diag fw events callbacks
  * @ctx: handle to wmi
@@ -2653,27 +2654,6 @@ static void wmi_control_diag_rx(void *ctx, HTC_PACKET *htc_packet)
 
 	wmi_process_control_rx(wmi_handle, evt_buf);
 }
-
-#elif defined(WLAN_DIAG_AND_DBR_OVER_SEPARATE_CE)
-static void wmi_control_diag_rx(void *ctx, HTC_PACKET *htc_packet)
-{
-	struct wmi_soc *soc = (struct wmi_soc *)ctx;
-	struct wmi_unified *wmi_handle;
-	wmi_buf_t evt_buf;
-
-	evt_buf = (wmi_buf_t)htc_packet->pPktContext;
-
-	wmi_handle = wmi_get_pdev_ep(soc, htc_packet->Endpoint);
-
-	if (!wmi_handle) {
-		wmi_err("unable to get wmi_handle for diag event end point id:%d", htc_packet->Endpoint);
-		qdf_nbuf_free(evt_buf);
-		return;
-	}
-
-	wmi_process_control_rx(wmi_handle, evt_buf);
-}
-
 #endif
 
 #ifdef WLAN_FEATURE_WMI_SEND_RECV_QMI