diff --git a/ACL_PyTorch/built-in/audio/whisperx/infer.py b/ACL_PyTorch/built-in/audio/whisperx/infer.py
index 5396ddcd97e914652d07eb7afecc624f8bb54130..c80a6b3bc3a3a5c2e4b6b712dd11f4e4b504423e 100644
--- a/ACL_PyTorch/built-in/audio/whisperx/infer.py
+++ b/ACL_PyTorch/built-in/audio/whisperx/infer.py
@@ -90,7 +90,7 @@ if __name__ == '__main__':
     audio_sample = load_audio(args.audio_path)
 
     data_path = f'{args.speech_path}/1919/142785'
-    audio_files = collect_audio_files([data_path])[args.num_audio_files]
+    audio_files = collect_audio_files([data_path])[:args.num_audio_files]
     
     def get_audio(audio_file):
         return load_audio(audio_file)