Fix bug in FFT