Fixing a colorbar issue.