diff --git a/neuromllite/gui/NMLliteUI.py b/neuromllite/gui/NMLliteUI.py
index 389b9b2..deb8d92 100644
--- a/neuromllite/gui/NMLliteUI.py
+++ b/neuromllite/gui/NMLliteUI.py
@@ -1044,7 +1044,7 @@ def replotSimResults(self):
         ax_heatmap.clear()
 
         if len(heat_array) > 0:
-            cm = matplotlib.cm.get_cmap("jet")
+            cm = matplotlib.pyplot.get_cmap("jet")
             hm = ax_heatmap.pcolormesh(heat_array, cmap=cm)
             # cbar = ax_heatmap.colorbar(im)