diff --git a/egs/librilight/SSL/prepare.sh b/egs/librilight/SSL/prepare.sh
index 160c66c7da..fd1037eea6 100755
--- a/egs/librilight/SSL/prepare.sh
+++ b/egs/librilight/SSL/prepare.sh
@@ -86,15 +86,15 @@ if [ $stage -le 4 ] && [ $stop_stage -ge 4 ]; then
     wget https://dl.fbaipublicfiles.com/hubert/hubert_base_ls960_L9_km500.bin -P download
   fi
   if [ ! -e data/kmeans/.extract_small.done ]; then
-    ./local/extract_kmeans.py --subset small
+    ./local/discretize_with_s3tokenizer.py --subset small
     touch data/kmeans/.extract_small.done
   fi
   if [ ! -e data/kmeans/.extract_medium.done ]; then
-    ./local/extract_kmeans.py --subset medium
+    ./local/discretize_with_s3tokenizer.py --subset medium
     touch data/kmeans/.extract_medium.done
   fi
   if [ ! -e data/kmeans/.extract_large.done ]; then
-    ./local/extract_kmeans.py --subset large
+    ./local/discretize_with_s3tokenizer.py --subset large
     touch data/kmeans/.extract_large.done
   fi
 fi