diff --git a/settings.gradle b/settings.gradle
index 9796a4950011a9832e1da486fbb7363c403cd6eb..a8da5d5dfd548bc9c359c7c21902acce224e77a4 100644
--- a/settings.gradle
+++ b/settings.gradle
@@ -12,5 +12,4 @@ include 'uc3-application'
 include 'uc4-workload-generator'
 include 'uc4-application'
 
-include 'workload-generator-common'
-include 'test-workload-generator'
+include 'workload-generator-common'
\ No newline at end of file
diff --git a/workload-generator-common/src/main/java/common/generators/KafkaWorkloadGenerator.java b/workload-generator-common/src/main/java/common/generators/KafkaWorkloadGenerator.java
index 66553cc1982397d329775b20a42db4336395e9f5..44327083dbd6236ffd2b5b16d45eb03c6f346a08 100644
--- a/workload-generator-common/src/main/java/common/generators/KafkaWorkloadGenerator.java
+++ b/workload-generator-common/src/main/java/common/generators/KafkaWorkloadGenerator.java
@@ -19,6 +19,7 @@ public class KafkaWorkloadGenerator<T extends IMonitoringRecord> extends Workloa
   /**
    * Create a new workload generator.
    *
+   * @param zooKeeper a reference to the ZooKeeper instance.
    * @param keySpace the key space to generate the workload for.
    * @param threads tha amount of threads to use per instance.
    * @param period the period how often a message is generated for each key specified in the