diff --git a/AMI_run.sh b/AMI_run.sh
index ab9ed68..18798ec 100755
--- a/AMI_run.sh
+++ b/AMI_run.sh
@@ -1,6 +1,6 @@
 #!/usr/bin/env bash
 
-stage=1
+stage=0
 AMIcorpus=example/ami/amicorpus # If you already have amicorpus, please set the path to that location and skip stage 0.
 oracle_rttm_path=example/ami/rttm
 testset_path=example/ami/ami_testset
@@ -17,11 +17,11 @@ if [ $stage -le 1 ]; then
     for audio_path in $(find $AMIcorpus/*/audio -name *.wav)
     do
         audio_name=`basename $audio_path`
-        # echo  $audio_path
+        new_audio_name=$(basename "$audio_name" .Mix-Headset.wav).wav
         for test_name in $(cat example/ami/ami_testset_list.txt)
         do
             if [ $audio_name == $test_name ]; then
-                ln -s `pwd`/$audio_path $testset_path
+                ln -s `pwd`/$audio_path $testset_path/$new_audio_name
             fi
         done
     done
diff --git a/setup.py b/setup.py
index 739afca..3ca2d23 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name="RTTManz",
-    version="0.3.0",
+    version="0.3.1",
     author="Yu-Sen Cheng (DTDwind)",
     description="A simple Python package for analyzing the necessary data in Speaker Diarization using oracle RTTM files and audio files.",
     keywords="analyze rttm",