Fix fft output a bit