diff --git a/distribution-diagram/pom.xml b/distribution-diagram/pom.xml
index 7222d0362..fb3c4049a 100644
--- a/distribution-diagram/pom.xml
+++ b/distribution-diagram/pom.xml
@@ -48,6 +48,10 @@
             <groupId>${project.groupId}</groupId>
             <artifactId>powsybl-single-line-diagram-core</artifactId>
         </dependency>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>powsybl-diagram-test</artifactId>
+        </dependency>
     </dependencies>
 
     <profiles>