Fixing little detail in heatmap JS