diff --git a/setup.py b/setup.py
index c864d77b..72f5c41b 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setuptools.setup(
     name="wfl",
-    version="0.2.4",
+    version="0.2.5",
     packages=setuptools.find_packages(exclude=["tests"]),
     install_requires=["click>=7.0", "numpy", "ase>=3.22.1", "pyyaml", "spglib", "docstring_parser",
                       "expyre-wfl @ https://github.com/libAtoms/ExPyRe/tarball/main",
diff --git a/wfl/fit/error.py b/wfl/fit/error.py
index 0618738a..daeaf50b 100755
--- a/wfl/fit/error.py
+++ b/wfl/fit/error.py
@@ -5,7 +5,7 @@
 import numpy as np
 
 from matplotlib.figure import Figure
-from matplotlib.cm import get_cmap
+from matplotlib.pyplot import get_cmap
 
 
 def calc(inputs, calc_property_prefix, ref_property_prefix,