fix color heat map