address comment

cosmos · Dec 3, 2024 · c721c5a · c721c5a
1 parent 434c66d
commit c721c5a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/integration/v2/services.go b/tests/integration/v2/services.go
@@ -76,10 +76,11 @@ type integrationContext struct {
 
 func SetHeaderInfo(ctx context.Context, h header.Info) context.Context {
 	iCtx, ok := ctx.Value(contextKey).(*integrationContext)
-	if ok {
-		iCtx.header = h
+	if !ok {
+		return ctx
 	}
-	return ctx
+	iCtx.header = h
+	return context.WithValue(ctx, contextKey, iCtx)
 }
 
 func HeaderInfoFromContext(ctx context.Context) header.Info {