diff --git a/src/NServiceBus.Extensions.IntegrationTesting/EndpointFixture.cs b/src/NServiceBus.Extensions.IntegrationTesting/EndpointFixture.cs
index 7beef2c..94b26d9 100644
--- a/src/NServiceBus.Extensions.IntegrationTesting/EndpointFixture.cs
+++ b/src/NServiceBus.Extensions.IntegrationTesting/EndpointFixture.cs
@@ -100,8 +100,8 @@ private static async Task<ObservedMessageContexts> ExecuteAndWait<TMessageContex
 
                         break;
                     case NsbActivityNames.InvokeHandlerActivityName:
-                        var handlerContext = activitySource.Parent.GetTagItem("testing.invoke.handler.context") as IInvokeHandlerContext;
-                     
+                        var handlerContext = (activitySource.GetTagItem("testing.invoke.handler.context") ?? activitySource.Parent?.GetTagItem("testing.invoke.handler.context")) as IInvokeHandlerContext;
+
                         invokeHandlerContexts.Add(handlerContext);
 
                         if (handlerContext is TMessageContext ctx3 && predicate(ctx3))