diff --git a/zebra/zebra_dplane.c b/zebra/zebra_dplane.c
index a98a1bce79..6b3d24aeb3 100644
--- a/zebra/zebra_dplane.c
+++ b/zebra/zebra_dplane.c
@@ -729,6 +729,14 @@ uint32_t dplane_ctx_get_notif_provider(const struct zebra_dplane_ctx *ctx)
 	return ctx->zd_notif_provider;
 }
 
+void dplane_ctx_set_notif_provider(struct zebra_dplane_ctx *ctx,
+				       uint32_t id)
+{
+	DPLANE_CTX_VALID(ctx);
+
+	ctx->zd_notif_provider = id;
+}
+
 void dplane_ctx_set_type(struct zebra_dplane_ctx *ctx, int type)
 {
 	DPLANE_CTX_VALID(ctx);
diff --git a/zebra/zebra_dplane.h b/zebra/zebra_dplane.h
index 75b7748cb8..60fd05e22d 100644
--- a/zebra/zebra_dplane.h
+++ b/zebra/zebra_dplane.h
@@ -202,6 +202,8 @@ void dplane_ctx_set_vrf(struct zebra_dplane_ctx *ctx, vrf_id_t vrf);
 vrf_id_t dplane_ctx_get_vrf(const struct zebra_dplane_ctx *ctx);
 
 bool dplane_ctx_is_from_notif(const struct zebra_dplane_ctx *ctx);
+void dplane_ctx_set_notif_provider(struct zebra_dplane_ctx *ctx,
+				   uint32_t id);
 uint32_t dplane_ctx_get_notif_provider(const struct zebra_dplane_ctx *ctx);
 
 /* Accessors for route update information */