diff --git a/test/unit/org/apache/cassandra/service/writes/thresholds/WriteWarningsSnapshotTest.java b/test/unit/org/apache/cassandra/service/writes/thresholds/WriteWarningsSnapshotTest.java
index c22ec7df3cf3..7da9c18c4107 100644
--- a/test/unit/org/apache/cassandra/service/writes/thresholds/WriteWarningsSnapshotTest.java
+++ b/test/unit/org/apache/cassandra/service/writes/thresholds/WriteWarningsSnapshotTest.java
@@ -175,11 +175,10 @@ private static Gen<WriteWarningsSnapshot> nonEmpty()
 
     private static Gen<WriteThresholdCounter> counter()
     {
-        Gen<Boolean> isEmpty = SourceDSL.booleans().all();
         Constraint maxValue = Constraint.between(1, Long.MAX_VALUE);
         Gen<WriteThresholdCounter> gen = rs ->
         {
-            if (isEmpty.generate(rs))
+            if (rs.next(Constraint.between(0, 3)) == 0)
                 return WriteThresholdCounter.empty();
             Map<TableId, Long> values = new HashMap<>();
             values.put(TABLE1, rs.next(maxValue));