Fixed bug w/ batch fft