Fix plot colorbar