diff --git a/post_processing/rayleigh_diagnostics.py b/post_processing/rayleigh_diagnostics.py
index 5baa5908..120734ab 100644
--- a/post_processing/rayleigh_diagnostics.py
+++ b/post_processing/rayleigh_diagnostics.py
@@ -2489,6 +2489,8 @@ def plot_azav(fig,ax,field,radius,costheta,sintheta,r_bcz=0.71,mini=-1,maxi=-1,m
     levs=mini+np.linspace(1,nlevs,nlevs)/float(nlevs)*(maxi-mini)
     ax.contour(yr,xr,field,colors='w',levels=levs)
 
+    return img
+
 def streamfunction(vr,vt,r,cost,order=0):
     """------------------------------------------------------------
     This routine takes as input a divergenceless axisymmetric