Make sure heatmap is cancellable