diff --git a/environment.yml b/environment.yml
index ca6df49..0c7a2f6 100644
--- a/environment.yml
+++ b/environment.yml
@@ -9,6 +9,6 @@ dependencies:
   - dask[complete]
   - pytest
   - matplotlib
-  - graphviz
+  - python-graphviz
   - jupyterlab