heat map bug fix