Fixing a bug with 3D heatmap.