diff --git a/whisper_live/transcriber_tensorrt.py b/whisper_live/transcriber_tensorrt.py
index 8634a8f..a36bb72 100644
--- a/whisper_live/transcriber_tensorrt.py
+++ b/whisper_live/transcriber_tensorrt.py
@@ -11,7 +11,7 @@ import numpy as np
 from whisper.tokenizer import get_tokenizer
 from whisper_live.tensorrt_utils import (mel_filters, store_transcripts,
                            write_error_stats, load_audio_wav_format,
-                           pad_or_trim)
+                           pad_or_trim, load_audio)
 
 import tensorrt_llm
 import tensorrt_llm.logger as logger
@@ -337,4 +337,4 @@ def decode_wav_file(
     if normalizer:
         prediction = normalizer(prediction)
     
-    return prediction.strip()
\ No newline at end of file
+    return prediction.strip()