ENH: fixing bugs colorbar