Fix plot 3 (and corresponding plot 4)