Fix axes for heatmaps