diff --git a/hiddenlayer/canvas.py b/hiddenlayer/canvas.py
index <HASH>..<HASH> 100644
--- a/hiddenlayer/canvas.py
+++ b/hiddenlayer/canvas.py
@@ -10,7 +10,6 @@ Licensed under the MIT License
 import itertools
 import math
 import numpy as np
-from graphviz import Digraph
 import matplotlib
 import matplotlib.pyplot as plt
 import IPython.display
diff --git a/hiddenlayer/graph.py b/hiddenlayer/graph.py
index <HASH>..<HASH> 100644
--- a/hiddenlayer/graph.py
+++ b/hiddenlayer/graph.py
@@ -12,7 +12,6 @@ import os
 import re
 from random import getrandbits
 import numpy as np
-from graphviz import Digraph
 
 
 THEMES = {
@@ -304,6 +303,8 @@ class Graph():
 
         Returns a GraphViz Digraph object.
         """
+        from graphviz import Digraph
+
         # Build GraphViz Digraph
         dot = Digraph()
         dot.attr("graph",
Allow importing when Graphviz is not installed

Import GraphViz only when needed so other features, such as plotting, can be used even if GraphViz is not available.