diff --git a/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/LoadGenerator.java b/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/LoadGenerator.java
index b4f2d9ab4aedd6c030e3d99a6f0848432e7361dc..ed5dd116c12ce76d800e904c8534282a0823239b 100644
--- a/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/LoadGenerator.java
+++ b/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/LoadGenerator.java
@@ -150,8 +150,8 @@ public final class LoadGenerator {
                     kafkaInputTopic,
                     schemaRegistryUrl,
                     kafkaProperties)
-                .forConstantValue(value))
-                    .setThreads(threads));
+                .forConstantValue(value)))
+        .withThreads(threads);
   }
 
 }