diff --git a/flavio/plots/plotfunctions.py b/flavio/plots/plotfunctions.py
index 6dd7b0e0..76f9b458 100644
--- a/flavio/plots/plotfunctions.py
+++ b/flavio/plots/plotfunctions.py
@@ -579,6 +579,7 @@ def contour(x, y, z, levels,
     CS = ax.contour(x, y, z, levels=levels, **_contour_args)
     if label is not None:
         CS.collections[0].set_label(label)
+    return CS
 
 
 def flavio_branding(x=0.8, y=0.94, version=True):