Fix bug with heatmap overlay