diff --git a/docker/Dockerfile b/docker/Dockerfile
index 75443cc..778a867 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -5,7 +5,6 @@ ENV DEB_PYTHON_INSTALL_LAYOUT=deb_system
 WORKDIR /tmp/
 
 COPY . ./dolfinx-adjoint
+RUN apt update && apt install -y graphviz-dev &&  apt-get clean
 RUN python3 -m pip install ./dolfinx-adjoint[all]
-
-
-RUN rm -rf /tmp
+RUN rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*