diff --git a/tests/BotSharp.LLM.Tests/Core/TestAgentService.cs b/tests/BotSharp.LLM.Tests/Core/TestAgentService.cs
index e9d306af..6716519a 100644
--- a/tests/BotSharp.LLM.Tests/Core/TestAgentService.cs
+++ b/tests/BotSharp.LLM.Tests/Core/TestAgentService.cs
@@ -1,4 +1,4 @@
-using BotSharp.Abstraction.Agents;
+using BotSharp.Abstraction.Agents;
 using BotSharp.Abstraction.Agents.Enums;
 using BotSharp.Abstraction.Agents.Models;
 using BotSharp.Abstraction.Functions.Models;
@@ -26,7 +26,7 @@ namespace BotSharp.Plugin.Google.Core
             return Task.FromResult(new PagedItems<Agent>());
         }
 
-        public Task<List<IdName>> GetAgentOptions(List<string>? agentIds = null)
+        public Task<List<IdName>> GetAgentOptions(List<string>? agentIds = null, bool byName = false)
         {
             return Task.FromResult(new List<IdName> { new IdName(id: "1", name: "Fake Agent") });
         }