Fixed a heatmap bug