diff --git a/src/OrleansTestKit/OrleansTestKit.csproj b/src/OrleansTestKit/OrleansTestKit.csproj
index 1940da3..545b724 100644
--- a/src/OrleansTestKit/OrleansTestKit.csproj
+++ b/src/OrleansTestKit/OrleansTestKit.csproj
@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8"?>
 <Project Sdk="Microsoft.NET.Sdk">
 
   <PropertyGroup>
@@ -12,7 +12,7 @@
     <PackageTags>Orleans Cloud-Computing Actor-Model Actors Distributed-Systems C# .NET Test Testing</PackageTags>
     <PublishRepositoryUrl>true</PublishRepositoryUrl>
     <SymbolPackageFormat>snupkg</SymbolPackageFormat>
-    <Version>8.0.0</Version>
+    <Version>8.1.0</Version>
   </PropertyGroup>
 
   <PropertyGroup>
@@ -24,9 +24,9 @@
       <PrivateAssets>all</PrivateAssets>
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
-    <PackageReference Include="Microsoft.Orleans.Reminders" Version="8.0.0" />
-    <PackageReference Include="Microsoft.Orleans.Streaming" Version="8.0.0" />
-    <PackageReference Include="Microsoft.Orleans.Runtime" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Orleans.Reminders" Version="8.1.0" />
+    <PackageReference Include="Microsoft.Orleans.Streaming" Version="8.1.0" />
+    <PackageReference Include="Microsoft.Orleans.Runtime" Version="8.1.0" />
     <PackageReference Include="Microsoft.SourceLink.GitHub" Version="1.1.1">
       <IncludeAssets>runtime; build; native; contentfiles; analyzers</IncludeAssets>
       <PrivateAssets>all</PrivateAssets>
diff --git a/src/OrleansTestKit/Reminders/ReminderContextHolder.cs b/src/OrleansTestKit/Reminders/ReminderContextHolder.cs
index 73764f2..becd4b5 100644
--- a/src/OrleansTestKit/Reminders/ReminderContextHolder.cs
+++ b/src/OrleansTestKit/Reminders/ReminderContextHolder.cs
@@ -20,7 +20,7 @@ private ReminderContextHolder()
         _slim = new SemaphoreSlim(1);
         var assembly = typeof(GrainId).Assembly;
         var contextType = assembly.GetType(RuntimeNamespace)!;
-        _runtimeContextMethod = contextType.GetMethod(RuntimeMethod, BindingFlags.NonPublic | BindingFlags.Static, new Type[] { typeof(IGrainContext) })!;
+        _runtimeContextMethod = contextType.GetMethod(RuntimeMethod, BindingFlags.NonPublic | BindingFlags.Static)!;
     }
 
     public static ReminderContextHolder Instance =>
@@ -33,7 +33,7 @@ public void ReleaseReminderContext()
     {
         try
         {
-            _runtimeContextMethod.Invoke(null, new object?[] { null });
+            _runtimeContextMethod.Invoke(null, new object?[] { null, null });
         }
         finally
         {
@@ -44,6 +44,6 @@ public void ReleaseReminderContext()
     public async Task SetReminderContext(IGrainContext context, CancellationToken token = default)
     {
         await _slim.WaitAsync(token);
-        _runtimeContextMethod.Invoke(null, new object[] { context });
+        _runtimeContextMethod.Invoke(null, new object?[] { context, null });
     }
 }
diff --git a/src/OrleansTestKit/Utilities/RuntimeContextManager.cs b/src/OrleansTestKit/Utilities/RuntimeContextManager.cs
index bfe7bf1..5270ddf 100644
--- a/src/OrleansTestKit/Utilities/RuntimeContextManager.cs
+++ b/src/OrleansTestKit/Utilities/RuntimeContextManager.cs
@@ -5,28 +5,30 @@ namespace Orleans.TestKit.Reminders;
 
 internal sealed class RuntimeContextManager
 {
+    public delegate void SetExecutionContextAction(IGrainContext? newContext, out IGrainContext currentContext);
+
     private RuntimeContextManager()
     {
         var assembly = typeof(GrainId).Assembly;
         var contextType = assembly.GetType("Orleans.Runtime.RuntimeContext")!;
 
         SetExecutionContext = contextType
-            .GetMethod("SetExecutionContext", BindingFlags.NonPublic | BindingFlags.Static, new Type[] { typeof(IGrainContext) })!
-            .CreateDelegate<Action<IGrainContext?>>()
+            .GetMethod("SetExecutionContext", BindingFlags.NonPublic | BindingFlags.Static)!
+            .CreateDelegate<SetExecutionContextAction>()
         ;
     }
 
     public static RuntimeContextManager Instance { get; } = new RuntimeContextManager();
 
-    public Action<IGrainContext?> SetExecutionContext { get; }
+    public SetExecutionContextAction SetExecutionContext { get; }
 
     public static IDisposable StartExecutionContext(IGrainContext context) => new RuntimeContextScope(context);
 
-    public void ResetExecutionContext() => SetExecutionContext(null);
+    public void ResetExecutionContext() => SetExecutionContext(null, out _);
 
     private sealed class RuntimeContextScope : IDisposable
     {
-        public RuntimeContextScope(IGrainContext context) => Instance.SetExecutionContext(context);
+        public RuntimeContextScope(IGrainContext context) => Instance.SetExecutionContext(context, out _);
 
         public void Dispose() => Instance.ResetExecutionContext();
     }
diff --git a/test/OrleansTestKit.Tests/OrleansTestKit.Tests.csproj b/test/OrleansTestKit.Tests/OrleansTestKit.Tests.csproj
index 7c76e58..9dd68db 100644
--- a/test/OrleansTestKit.Tests/OrleansTestKit.Tests.csproj
+++ b/test/OrleansTestKit.Tests/OrleansTestKit.Tests.csproj
@@ -14,7 +14,7 @@
     <PackageReference Include="FluentAssertions" Version="6.12.0" />
     <PackageReference Include="Microsoft.CodeCoverage" Version="17.8.0" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.8.0" />
-    <PackageReference Include="Microsoft.Orleans.Sdk" Version="8.0.0" />
+    <PackageReference Include="Microsoft.Orleans.Sdk" Version="8.1.0" />
     <PackageReference Include="NSubstitute" Version="5.1.0" />
     <PackageReference Include="System.Configuration.ConfigurationManager" Version="7.0.0" />
     <PackageReference Include="xunit" Version="2.6.5" />