Minor fix to plot_fft