diff --git a/Dockerfile b/Dockerfile
index 8a9ba58..2a2ea87 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,7 +20,10 @@ LABEL org.opencontainers.image.title="Quantum Lab" \
       
 # OS requirements
 RUN apt-get update -yq \
-&& apt-get install python3-tk git -yq \
+&& apt-get install -yq \
+      python3-tk \
+      graphviz \
+      git \
 && apt-get dist-upgrade -yq \
 && apt-get clean -yq
 
diff --git a/build/requirements.txt b/build/requirements.txt
index 7cb1697..a5b478d 100644
--- a/build/requirements.txt
+++ b/build/requirements.txt
@@ -3,7 +3,6 @@ pylatexenc
 notebook
 purplecaffeine
 prototype-zne
-graphviz
 qiskit==1.0.2
 qiskit[visualization]
 qiskit_aer==0.14.1