fix colorbar bug