diff --git a/README.md b/README.md
index 2da0eda6..4ef5f3ae 100644
--- a/README.md
+++ b/README.md
@@ -106,7 +106,7 @@ Now the SHARK will use your locally build Torch-MLIR repo.
 ```shell
 python -m  shark.examples.shark_inference.resnet50_script --device="cpu" # Use gpu | vulkan
 # Or a pytest
-pytest tank/tf/hf_masked_lm/albert-base-v2_test.py::AlbertBaseModuleTest::test_module_static_cpu
+pytest tank/test_models.py -k "MiniLM"
 ```
 
 </details>
diff --git a/setup_venv.sh b/setup_venv.sh
index 7701289f..44c1cacf 100755
--- a/setup_venv.sh
+++ b/setup_venv.sh
@@ -71,18 +71,11 @@ else
   echo "${Red}OS not detected. Pray and Play"
 fi
 
-TM_HTML_URL="$(python3 -c "import urllib.request, json, sys; \
-	u=json.loads(urllib.request.urlopen('https://api.github.com/repos/llvm/torch-mlir/releases/latest').read().decode()).get('html_url', False);\
-	print(u) if u else sys.exit(1);")"
-TM_RELEASE_DIR=${TM_HTML_URL/"tag"/"expanded_assets"}
-echo "TM_HTML_URL=${TM_HTML_URL}"
-echo "TM_RELEASE_DIR=${TM_RELEASE_DIR}"
-
 # Upgrade pip and install requirements.
 $PYTHON -m pip install --upgrade pip || die "Could not upgrade pip"
 $PYTHON -m pip install --upgrade -r "$TD/requirements.txt"
 if [ "$torch_mlir_bin" = true ]; then
-  $PYTHON -m pip install --find-links ${TM_RELEASE_DIR} torch-mlir --extra-index-url ${TM_RELEASE_DIR}
+  $PYTHON -m pip install --pre torch-mlir -f https://llvm.github.io/torch-mlir/package-index/
   if [ $? -eq 0 ];then
     echo "Successfully Installed torch-mlir"
   else
@@ -117,7 +110,7 @@ if [[ ! -z "${IMPORTER}" ]]; then
   fi
 fi
 
-$PYTHON -m pip install -e . -f ${TM_RELEASE_DIR} -f https://github.com/${RUNTIME}/releases
+$PYTHON -m pip install -e . -f https://llvm.github.io/torch-mlir/package-index/ -f https://github.com/${RUNTIME}/releases
 
 if [[ $(uname -s) = 'Linux' && ! -z "${IMPORTER}" ]]; then
   $PYTHON -m pip uninstall -y torch torchvision