Fix fft_size calculation