diff --git a/nucleus/gnmi_transport.go b/nucleus/gnmi_transport.go
index fc736ac7adfa01cdd3e7f031d9554f70ff910ea4..d8d9e60abdf6a20426bd6f61d824c468983e02a0 100644
--- a/nucleus/gnmi_transport.go
+++ b/nucleus/gnmi_transport.go
@@ -90,6 +90,7 @@ func (g *Gnmi) Set(ctx context.Context, payload change.Payload) error {
 	if g.client == nil {
 		return &errors.ErrNilClient{}
 	}
+	ctx = gnmi.NewContext(ctx, g.config)
 	return g.applyDiff(ctx, payload)
 }