FIX : return fig not axes