From 47c35815640f482921e3b19638e589985146b9dd Mon Sep 17 00:00:00 2001
From: Peter Stangl <peterstangl@users.noreply.github.com>
Date: Fri, 15 Mar 2019 22:47:00 +0100
Subject: [PATCH] [plots] return CS in contour (#81)

---
 flavio/plots/plotfunctions.py | 1 +
 1 file changed, 1 insertion(+)

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):