From 138bc777dcdba4757f045df90f3afb9ade8dace3 Mon Sep 17 00:00:00 2001
From: Daniel Marbach <danielmarbach@users.noreply.github.com>
Date: Thu, 14 Aug 2025 18:24:05 +0200
Subject: [PATCH] Adjust to nullable context bag

---
 src/Transport/Sending/MessageDispatcher.cs                  | 2 +-
 src/Transport/Sending/NativeMessageCustomizationBehavior.cs | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Transport/Sending/MessageDispatcher.cs b/src/Transport/Sending/MessageDispatcher.cs
index 2de290b8..69a882ef 100644
--- a/src/Transport/Sending/MessageDispatcher.cs
+++ b/src/Transport/Sending/MessageDispatcher.cs
@@ -25,7 +25,7 @@ readonly OutgoingNativeMessageCustomizationAction
 
     public async Task Dispatch(TransportOperations outgoingMessages, TransportTransaction transaction, CancellationToken cancellationToken = default)
     {
-        _ = transaction.TryGet(out AzureServiceBusTransportTransaction azureServiceBusTransaction);
+        _ = transaction.TryGet<AzureServiceBusTransportTransaction>(out var azureServiceBusTransaction);
 
         var unicastTransportOperations = outgoingMessages.UnicastTransportOperations;
         var multicastTransportOperations = outgoingMessages.MulticastTransportOperations;
diff --git a/src/Transport/Sending/NativeMessageCustomizationBehavior.cs b/src/Transport/Sending/NativeMessageCustomizationBehavior.cs
index fda231ea..dcbe3bbb 100644
--- a/src/Transport/Sending/NativeMessageCustomizationBehavior.cs
+++ b/src/Transport/Sending/NativeMessageCustomizationBehavior.cs
@@ -12,7 +12,7 @@ sealed class NativeMessageCustomizationBehavior(bool isOutboxEnabled) : Behavior
 
     public override Task Invoke(IRoutingContext context, Func<Task> next)
     {
-        if (!context.Extensions.TryGet(CustomizationKey, out Action<ServiceBusMessage> customization))
+        if (!context.Extensions.TryGet<Action<ServiceBusMessage>>(CustomizationKey, out var customization))
         {
             return next();
         }