Heat map fix