diff --git a/audio_playback.py b/audio_playback.py
index 001e95e..17d909c 100644
--- a/audio_playback.py
+++ b/audio_playback.py
@@ -7,7 +7,7 @@ from scipy.io import wavfile
 from scipy.io.wavfile import write
 import subprocess
 import sounddevice
-
+from pydub import AudioSegment
 
 audio_path = os.path.join(folder_paths.get_input_directory(),"audio")