fixing heat map issues