One more small change to plot 4