diff --git a/src/openfe/tests/utils/test_atommapping_network_plotting.py b/src/openfe/tests/utils/test_atommapping_network_plotting.py
index 9e8086f88..d823b8d5d 100644
--- a/src/openfe/tests/utils/test_atommapping_network_plotting.py
+++ b/src/openfe/tests/utils/test_atommapping_network_plotting.py
@@ -1,9 +1,7 @@
-import importlib.resources
 import inspect
 from unittest import mock
 
 import matplotlib
-import matplotlib.figure
 import pytest
 from matplotlib import pyplot as plt
 
diff --git a/src/openfe/utils/atommapping_network_plotting.py b/src/openfe/utils/atommapping_network_plotting.py
index 13d899d66..98255c69b 100644
--- a/src/openfe/utils/atommapping_network_plotting.py
+++ b/src/openfe/utils/atommapping_network_plotting.py
@@ -10,7 +10,7 @@
 from rdkit import Chem
 
 from openfe import LigandNetwork, SmallMoleculeComponent
-from openfe.utils.custom_typing import MPL_MouseEvent
+from openfe.utils import requires_package
 from openfe.utils.network_plotting import Edge, GraphDrawing, Node
 
 
@@ -156,6 +156,7 @@ class AtomMappingNetworkDrawing(GraphDrawing):
     EdgeCls = AtomMappingEdge
 
 
+@requires_package("pygraphviz")
 def plot_atommapping_network(network: LigandNetwork):
     """Convenience method for plotting the atom mapping network