diff --git a/src/test/java/com/lambdatest/TestNGSmartUI.java b/src/test/java/com/lambdatest/TestNGSmartUI.java
index 70b43e8..5d4c2f6 100644
--- a/src/test/java/com/lambdatest/TestNGSmartUI.java
+++ b/src/test/java/com/lambdatest/TestNGSmartUI.java
@@ -37,6 +37,9 @@ public void setup(Method m, ITestContext ctx) throws MalformedURLException {
     caps.setCapability("name", m.getName() + " - " + this.getClass().getName());
     caps.setCapability("plugin", "git-testng");
     caps.setCapability("smartUI.project", "testng-smartui-web-project");
+    if(System.getenv("BUILD_NAME")!=null && System.getenv("BUILD_NAME")!=""){
+      caps.setCapability("smartUI.build",System.getenv("BUILD_NAME"));
+    }
     caps.setCapability("selenium_version", "4.8.0");
 
     if (githubURL != null) {