diff --git a/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/communication/zookeeper/WorkloadDistributor.java b/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/communication/zookeeper/WorkloadDistributor.java
index 2249abcbcb1071cf880b2ee80f5d41f2b3dab463..832d575a90ada5a16079783ba507cd5c6143fd00 100644
--- a/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/communication/zookeeper/WorkloadDistributor.java
+++ b/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/communication/zookeeper/WorkloadDistributor.java
@@ -14,7 +14,7 @@ import org.apache.zookeeper.Watcher.Event.EventType;
 import org.apache.zookeeper.data.Stat;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
-import theodolite.commons.workloadgeneration.dimensions.KeySpace;
+import theodolite.commons.workloadgeneration.KeySpace;
 import theodolite.commons.workloadgeneration.functions.BeforeAction;
 import theodolite.commons.workloadgeneration.misc.WorkloadDefinition;
 import theodolite.commons.workloadgeneration.misc.ZooKeeper;
@@ -22,6 +22,7 @@ import theodolite.commons.workloadgeneration.misc.ZooKeeper;
 /**
  * The central class responsible for distributing the workload through all workload generators.
  */
+@Deprecated
 public class WorkloadDistributor {
 
   private static final Logger LOGGER = LoggerFactory.getLogger(WorkloadDistributor.class);
diff --git a/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/misc/ZooKeeper.java b/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/misc/ZooKeeper.java
index a80490600ad9c9c22c198fc76b6d9f73bdc30584..56b1438a46c1cf60594a2282b3a4778063e29737 100644
--- a/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/misc/ZooKeeper.java
+++ b/benchmarks/workload-generator-commons/src/main/java/theodolite/commons/workloadgeneration/misc/ZooKeeper.java
@@ -3,6 +3,7 @@ package theodolite.commons.workloadgeneration.misc;
 /**
  * Wrapper for connection information for ZooKeeper.
  */
+@Deprecated
 public class ZooKeeper {
 
   private final String host;