diff --git a/yt_idv/opengl_support.py b/yt_idv/opengl_support.py
index 0d37647..06d5a2d 100644
--- a/yt_idv/opengl_support.py
+++ b/yt_idv/opengl_support.py
@@ -15,10 +15,10 @@
 
 from contextlib import ExitStack, contextmanager
 
+import matplotlib.pyplot as plt
 import numpy as np
 import traitlets
 import traittypes
-from matplotlib import colormaps as cm
 from OpenGL import GL
 
 # Set up a mapping from numbers to names
@@ -221,7 +221,7 @@ def __init__(self, *args, **kwargs):
     @traitlets.validate("colormap_name")
     def _validate_name(self, proposal):
         try:
-            cm.get_cmap(proposal["value"])
+            plt.get_cmap(proposal["value"])
         except ValueError:
             raise traitlets.TraitError(
                 "Colormap name needs to be known by" "matplotlib"
@@ -230,7 +230,7 @@ def _validate_name(self, proposal):
 
     @traitlets.observe("colormap_name")
     def _observe_colormap_name(self, change):
-        cmap = cm.get_cmap(change["new"])
+        cmap = plt.get_cmap(change["new"])
         cmap_vals = np.array(cmap(np.linspace(0, 1, 256)), dtype="f4")
         self.data = cmap_vals