diff --git a/Dockerfile.acsmchain b/Dockerfile.acsmchain
index 6995f63..eb0432e 100644
--- a/Dockerfile.acsmchain
+++ b/Dockerfile.acsmchain
@@ -39,6 +39,7 @@ RUN conda run -n $ENV_NAME pip install -r /acsmdc/requirements.txt
 VOLUME ["/acsmdc/data", "/acsmdc/figures", "/acsmdc/pipelines/params"]
 
 # Add JupyterLab
+RUN pip install graphviz
 RUN pip install --no-cache-dir jupyterlab
 #RUN pip install pipx
 #RUN pipx install renku
diff --git a/env_setup.sh b/env_setup.sh
index a2c67e2..068b724 100644
--- a/env_setup.sh
+++ b/env_setup.sh
@@ -32,6 +32,7 @@ fi
 echo "Installing additional pip packages..."
 pip install pybis==1.35 igor2 ipykernel sphinx
 pip install dash dash-bootstrap-components
+pip install graphviz
 
 # Check if pip installations were successful
 if [ $? -ne 0 ]; then