fix the heatmap call and you are done