Fix heat_map application