Fix fisrt display of heatmap