diff --git a/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs b/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs
index ce5d03b6..346f2f23 100644
--- a/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs
+++ b/src/Infrastructure/BotSharp.Abstraction/Browsing/IWebDriverHook.cs
@@ -5,4 +5,5 @@ namespace BotSharp.Abstraction.Browsing;
 public interface IWebDriverHook
 {
     Task<List<string>> GetUploadFiles(MessageInfo message);
+    Task OnLocateElement(MessageInfo message, string content);
 }
diff --git a/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs b/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs
index e70d7f27..b4b2eb2f 100644
--- a/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs
+++ b/src/Plugins/BotSharp.Plugin.WebDriver/Drivers/PlaywrightDriver/PlaywrightWebDriver.LocateElement.cs
@@ -114,6 +114,11 @@ public partial class PlaywrightWebDriver
             // fix if html has &
             result.Body = HttpUtility.HtmlDecode(html);
             result.IsSuccess = true;
+            var hooks = _services.GetServices<IWebDriverHook>();
+            foreach (var hook in hooks)
+            {
+                await hook.OnLocateElement(message, result.Body);
+            }
         }
         else if (count > 1)
         {
diff --git a/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs b/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs
index d30048c8..e2e3d34f 100644
--- a/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs
+++ b/src/Plugins/BotSharp.Plugin.WebDriver/UtilFunctions/UtilWebLocateElementFn.cs
@@ -30,6 +30,7 @@ public class UtilWebLocateElementFn : IFunctionCallback
             MessageId = message.MessageId,
             ContextId = webDriverService.GetMessageContext(message)
         };
+        browser.SetServiceProvider(_services);
         var result = await browser.LocateElement(msg, locatorArgs);
 
         message.Content = $"Locating element {(result.IsSuccess ? "success" : "failed")}. ";