Minor fix for plot 4