diff --git a/core/cds/src/cds_api.c b/core/cds/src/cds_api.c
index 232918eac3da..3222cc9fabaf 100644
--- a/core/cds/src/cds_api.c
+++ b/core/cds/src/cds_api.c
@@ -193,6 +193,7 @@ QDF_STATUS cds_init(void)
 	qdf_register_self_recovery_callback(__cds_trigger_recovery);
 	qdf_register_fw_down_callback(cds_is_fw_down);
 	qdf_register_ssr_protect_callbacks(cds_ssr_protect, cds_ssr_unprotect);
+	qdf_register_recovering_state_query_callback(cds_is_driver_recovering);
 
 	return QDF_STATUS_SUCCESS;
 
@@ -213,7 +214,7 @@ void cds_deinit(void)
 	QDF_BUG(gp_cds_context);
 	if (!gp_cds_context)
 		return;
-
+	qdf_register_recovering_state_query_callback(NULL);
 	qdf_register_ssr_protect_callbacks(NULL, NULL);
 	qdf_register_fw_down_callback(NULL);
 	qdf_register_self_recovery_callback(NULL);