diff --git a/src/Plugins/BotSharp.Plugin.ChatHub/ChatStreamMiddleware.cs b/src/Plugins/BotSharp.Plugin.ChatHub/ChatStreamMiddleware.cs
index 78e6d9e5..23ee7a10 100644
--- a/src/Plugins/BotSharp.Plugin.ChatHub/ChatStreamMiddleware.cs
+++ b/src/Plugins/BotSharp.Plugin.ChatHub/ChatStreamMiddleware.cs
@@ -1,4 +1,3 @@
-using Azure;
 using BotSharp.Abstraction.Realtime;
 using BotSharp.Abstraction.Realtime.Models;
 using Microsoft.AspNetCore.Http;
@@ -52,6 +51,7 @@ public class ChatStreamMiddleware
     {
         var hub = services.GetRequiredService<IRealtimeHub>();
         var conn = hub.SetHubConnection(conversationId);
+        conn.CurrentAgentId = agentId;
 
         // load conversation and state
         var convService = services.GetRequiredService<IConversationService>();