diff --git a/Pipfile b/Pipfile
index 4a08c49b..7c43d844 100644
--- a/Pipfile
+++ b/Pipfile
@@ -7,6 +7,7 @@ name = "pypi"
 networkx = "*"
 numpy = "*"
 scikit-learn = "*"
+matplotlib = "*"
 
 [dev-packages]
 pytest = "*"
@@ -20,4 +21,4 @@ sphinx-rtd-theme = "0.5.2"
 [extras]
 ray = "*"
 shap = "0.44.1"
-xarray = "*"
+xarray = "*"
\ No newline at end of file
diff --git a/setup.py b/setup.py
index dbe5a1fa..af6a678a 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@
 KEYWORDS = ["hierarchical classification"]
 DACS_SOFTWARE = "https://gitlab.com/dacs-hpi"
 # What packages are required for this module to be executed?
-REQUIRED = ["networkx", "numpy", "scikit-learn", "scipy<1.13"]
+REQUIRED = ["networkx", "numpy", "scikit-learn", "scipy<1.13", "matplotlib"]
 
 # What packages are optional?
 # 'fancy feature': ['django'],}