diff --git a/grill/views/_graph.py b/grill/views/_graph.py
index 3c0bcbc7..baef2a17 100644
--- a/grill/views/_graph.py
+++ b/grill/views/_graph.py
@@ -592,7 +592,7 @@ def _load_graph(self, graph):
             return
 
         try:  # exit early if pygraphviz is not installed, needed for positions
-            positions = drawing.nx_agraph.graphviz_layout(graph, prog='dot')
+            positions = drawing.nx_pydot.graphviz_layout(graph, prog='dot')
         except ImportError as exc:
             message = str(exc)
             print(message)
@@ -805,7 +805,7 @@ def _subgraph_dot_path(self, node_indices: tuple):
 
         fd, fp = tempfile.mkstemp()
         try:
-            nx.nx_agraph.write_dot(subgraph, fp)
+            nx.nx_pydot.write_dot(subgraph, fp)
         except ImportError as exc:
             error = f"{exc}\n\n{_DOT_ENVIRONMENT_ERROR}"
         else: