diff --git a/dd-trace-java.gradle b/dd-trace-java.gradle
index 10ee3cb67e..a7fe62deba 100644
--- a/dd-trace-java.gradle
+++ b/dd-trace-java.gradle
@@ -2,7 +2,7 @@ plugins {
   id 'io.franzbecker.gradle-lombok' version '1.14'
   id 'com.jfrog.artifactory' version '4.8.1'
   id 'com.jfrog.bintray' version '1.8.4'
-  id 'org.unbroken-dome.test-sets' version '1.5.2'
+  id 'org.unbroken-dome.test-sets' version '2.1.1'
 
   id 'com.gradle.build-scan' version '1.16' // 2.0+ requires gradle 5
   // Not applying google java format by default because it gets confused by stray java build