Fix colorbar not consistent issue