diff --git a/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamEnrichedStatusTests.java b/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamEnrichedStatusTests.java
index acd8cd1a6add..993db1096aac 100644
--- a/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamEnrichedStatusTests.java
+++ b/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamEnrichedStatusTests.java
@@ -74,7 +74,7 @@ public class ReindexDataStreamEnrichedStatusTests extends AbstractWireSerializin
     }
 
     private Set<String> randomSet(int minSize) {
-        return randomSet(minSize, 100, () -> randomAlphaOfLength(50));
+        return randomSet(minSize, Math.max(minSize, 100), () -> randomAlphaOfLength(50));
     }
 
     private List<Tuple<String, Exception>> randomErrorList() {
diff --git a/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamStatusTests.java b/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamStatusTests.java
index 47e2d02bee3b..ad47eb6a23cd 100644
--- a/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamStatusTests.java
+++ b/x-pack/plugin/migrate/src/test/java/org/elasticsearch/xpack/migrate/task/ReindexDataStreamStatusTests.java
@@ -70,7 +70,7 @@ public class ReindexDataStreamStatusTests extends AbstractWireSerializingTestCas
     }
 
     private Set<String> randomSet(int minSize) {
-        return randomSet(minSize, 100, () -> randomAlphaOfLength(50));
+        return randomSet(minSize, Math.max(minSize, 100), () -> randomAlphaOfLength(50));
     }
 
     private List<Tuple<String, Exception>> randomErrorList() {