@@ -57,7 +57,10 @@ public async Task ChangeListValue(Agent agent, BrowsingContextIn context, string
         }
 
         var driverService = _services.GetRequiredService<WebDriverService>();
-        var htmlElementContextOut = await driverService.LocateElement(agent, string.Join("", str), context.ElementName, messageId);
+        var htmlElementContextOut = await driverService.LocateElement(agent, 
+            string.Join("", str), 
+            context.ElementName, 
+            messageId);
         ILocator element = Locator(htmlElementContextOut);
 
         try
 
@@ -1,11 +1,11 @@
-using BotSharp.Plugin.WebDriver.Services;
-
 namespace BotSharp.Plugin.WebDriver.Drivers.PlaywrightDriver;
 
 public partial class PlaywrightWebDriver
 {
     public async Task ClickElement(Agent agent, BrowsingContextIn context, string messageId)
     {
+        var driverService = _services.GetRequiredService<WebDriverService>();
+
         // Retrieve the page raw html and infer the element path
         var body = await _instance.Page.QuerySelectorAsync("body");
 
@@ -23,18 +23,19 @@ public async Task ClickElement(Agent agent, BrowsingContextIn context, string me
             var text = await btn.TextContentAsync();
             var name = await btn.GetAttributeAsync("name");
             var id = await btn.GetAttributeAsync("id");
-            str.Add($"<button name='{name}' id='{id}'>{text}</button>");
+            str.Add(driverService.AssembleMarkup("button", new MarkupProperties
+            {
+                Id = id,
+                Name = name,
+                Text = text
+            }));
         }
 
-        var driverService = _services.GetRequiredService<WebDriverService>();
-        var htmlElementContextOut = await driverService.LocateElement(agent, string.Join("", str), context.ElementName, messageId);
-
-        var tags = await _instance.Page.QuerySelectorAllAsync(htmlElementContextOut.TagName);
-        var button = tags[htmlElementContextOut.Index];
-        if (button.AsElement() == null)
-        {
-            throw new Exception($"Can't find web element {context.ElementName}");
-        }
-        await button.ClickAsync();
+        var htmlElementContextOut = await driverService.LocateElement(agent, 
+            string.Join("", str), 
+            context.ElementName, 
+            messageId);
+        ILocator element = Locator(htmlElementContextOut);
+        await element.ClickAsync();
     }
 }
@@ -42,7 +42,10 @@ public async Task InputUserText(Agent agent, BrowsingContextIn context, string m
             }));
         }
 
-        var htmlElementContextOut = await driverService.LocateElement(agent, string.Join("", str), context.ElementName, messageId);
+        var htmlElementContextOut = await driverService.LocateElement(agent, 
+            string.Join("", str), 
+            context.ElementName, 
+            messageId);
         ILocator element = Locator(htmlElementContextOut);
 
         try
 
@@ -2,8 +2,8 @@ namespace BotSharp.Plugin.WebDriver.Models;
 
 internal class MarkupProperties
 {
-    public string Id { get; set; }
-    public string Name { get; set; }
-    public string Type { get; set; }
-    public string Text { get; set; }
+    public string? Id { get; set; }
+    public string? Name { get; set; }
+    public string? Type { get; set; }
+    public string? Text { get; set; }
 }
Original file line number	Diff line number	Diff line change
`@@ -2,8 +2,8 @@ namespace BotSharp.Plugin.WebDriver.Models;`
`2`	`2`
`3`	`3`	`internal class MarkupProperties`
`4`	`4`	`{`
`5`		`- public string Id { get; set; }`
`6`		`- public string Name { get; set; }`
`7`		`- public string Type { get; set; }`
`8`		`- public string Text { get; set; }`
	`5`	`+ public string? Id { get; set; }`
	`6`	`+ public string? Name { get; set; }`
	`7`	`+ public string? Type { get; set; }`
	`8`	`+ public string? Text { get; set; }`
`9`	`9`	`}`