diff --git a/build.gradle b/build.gradle
index d4b0993c5fc43ddbf15cec8dd8ad5d5557b2ee26..8cc8f591aaea1e005591b5faf21caa920cb267af 100644
--- a/build.gradle
+++ b/build.gradle
@@ -151,6 +151,9 @@ dependencies {
     implementation group: 'xalan', name: 'xalan', version: '2.7.2'
     dataImplementation group: 'org.apache.commons', name: 'commons-collections4', version: '4.4'
     compile group: 'org.apache.commons', name: 'commons-lang3', version: '3.0'
+    modelImplementation group: 'org.apache.commons', name: 'commons-collections4', version: '4.4'
+    implementation group: 'org.apache.commons', name: 'commons-collections4', version: '4.4'
+
 
     // from configs
     myBeaver group: 'net.sf.beaver', name: 'beaver-ant', version: '0.9.11'
diff --git a/src/main/jastadd/base/distribution/DistributedPN.jadd b/src/main/jastadd/base/distribution/DistributedPN.jadd
index 8ef3a73533f9ef349cef13542bbf7d3a212e5136..7102ae7ac24f459a65fe29b8145e75868fa9ee91 100644
--- a/src/main/jastadd/base/distribution/DistributedPN.jadd
+++ b/src/main/jastadd/base/distribution/DistributedPN.jadd
@@ -1,4 +1,5 @@
 import de.tudresden.inf.st.pnml.base.data.CommunicatorInformation;
+import org.apache.commons.collections4.MultiValuedMap;
 
 aspect PnDistribution {