On the Heatmap, make colors of the legend and of the graph match

I couldn't make these colors match:

<img width="700" alt="Image" src="https://github.com/user-attachments/assets/cf5a0981-7242-4331-bd05-19d3415c5f50" />