added ax=ax to fig.colorbar call, which is useful when packing figures with colorbars into subplots