Fixed a bug in the stft() function