Fix heat map good