diff --git a/polyfuzz/metrics.py b/polyfuzz/metrics.py
index 95eadb2..6e648f1 100644
--- a/polyfuzz/metrics.py
+++ b/polyfuzz/metrics.py
@@ -5,7 +5,7 @@
 import matplotlib.pyplot as plt
 from typing import Tuple, List, Mapping
 from matplotlib import gridspec
-from matplotlib.cm import get_cmap
+from matplotlib.pyplot import get_cmap
 from matplotlib.lines import Line2D