diff --git a/src/Infrastructure/BotSharp.Abstraction/Agents/Models/Agent.cs b/src/Infrastructure/BotSharp.Abstraction/Agents/Models/Agent.cs
index aa872566..431a78c5 100644
--- a/src/Infrastructure/BotSharp.Abstraction/Agents/Models/Agent.cs
+++ b/src/Infrastructure/BotSharp.Abstraction/Agents/Models/Agent.cs
@@ -1,6 +1,5 @@
 using BotSharp.Abstraction.Functions.Models;
 using BotSharp.Abstraction.Plugins.Models;
-using BotSharp.Abstraction.Routing.Enums;
 using BotSharp.Abstraction.Tasks.Models;
 
 namespace BotSharp.Abstraction.Agents.Models;
@@ -10,6 +9,7 @@ public class Agent
     public string Id { get; set; } = string.Empty;
     public string Name { get; set; } = string.Empty;
     public string Description { get; set; } = string.Empty;
+
     /// <summary>
     /// Agent Type
     /// </summary>
@@ -286,11 +286,11 @@ public class Agent
     }
 
     /// <summary>
-    /// Set agent mode: lazy or eager
+    /// Set agent routing mode: lazy or eager
     /// </summary>
     /// <param name="mode"></param>
     /// <returns></returns>
-    public Agent SetMode(string mode)
+    public Agent SetRoutingMode(string? mode)
     {
         Mode = mode;
         return this;
diff --git a/src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.UpdateAgent.cs b/src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.UpdateAgent.cs
index 50dbcad4..41174886 100644
--- a/src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.UpdateAgent.cs
+++ b/src/Infrastructure/BotSharp.Core/Agents/Services/AgentService.UpdateAgent.cs
@@ -95,7 +95,7 @@ public partial class AgentService
             clonedAgent.SetId(foundAgent.Id)
                        .SetName(foundAgent.Name)
                        .SetType(foundAgent.Type)
-                       .SetMode(foundAgent.Mode)
+                       .SetRoutingMode(foundAgent.Mode)
                        .SetIsPublic(foundAgent.IsPublic)
                        .SetDisabled(foundAgent.Disabled)
                        .SetDescription(foundAgent.Description)