fix colorbar plotting