diff --git a/README.md b/README.md
index e848e83..2cd6fe4 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-![Tests](https://github.com/pmlmodelling/ersem/workflows/build-ersem/badge.svg) 
+![Tests](https://img.shields.io/github/actions/workflow/status/pmlmodelling/ersem/ersem.yml?label=tests&style=flat-square) 
 [![Documentation Status](https://readthedocs.org/projects/ersem/badge/?version=latest)](https://ersem.readthedocs.io/en/latest/?badge=latest)
 
 # ERSEM
diff --git a/github-actions/pyfabm-ersem/pyfabm_tut.py b/github-actions/pyfabm-ersem/pyfabm_tut.py
index 2bcb64c..f1d79c3 100644
--- a/github-actions/pyfabm-ersem/pyfabm_tut.py
+++ b/github-actions/pyfabm-ersem/pyfabm_tut.py
@@ -1,4 +1,3 @@
-from matplotlib import cm
 import matplotlib.pylab as plt
 import numpy as np
 import os
@@ -70,7 +69,7 @@ def main():
     axes = plt.gca()
 
     # Set color map
-    cmap = cm.get_cmap('YlOrRd')
+    cmap = plt.colormaps.get_cmap('YlOrRd')
 
     # Plot
     plot = axes.pcolormesh(salinity_array,