Fixed issue in heat map