Fix an issue for discrete fourier transform test.