fix serious bug in calculate_ani.py heatmap output