diff --git a/setup_venv.sh b/setup_venv.sh
index 05198cac..b8fe33bb 100755
--- a/setup_venv.sh
+++ b/setup_venv.sh
@@ -6,7 +6,7 @@
 TD="$(cd $(dirname $0) && pwd)"
 VENV_DIR="$TD/shark.venv"
 if [ -z "$PYTHON" ]; then
-  PYTHON="$(which python)"
+  PYTHON="$(which python3)"
 fi
 
 echo "Setting up venv dir: $VENV_DIR"
@@ -25,11 +25,11 @@ source "$VENV_DIR/bin/activate" || die "Could not activate venv"
 # reference to the python executable from the venv.
 python -m pip install --upgrade pip || die "Could not upgrade pip"
 python -m pip install --upgrade -r "$TD/requirements.txt"
-pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir
-pip install --find-links https://github.com/NodLabs/SHARK/releases iree-compiler iree-runtime
-pip install git+https://github.com/pytorch/functorch.git
-pip install transformers
-pip install .
+python -m pip install --find-links https://github.com/llvm/torch-mlir/releases torch-mlir
+python -m pip install --find-links https://github.com/NodLabs/SHARK/releases iree-compiler iree-runtime
+python -m pip install git+https://github.com/pytorch/functorch.git
+python -m pip install transformers
+python -m pip install .
 
 Red=`tput setaf 1`          
 Green=`tput setaf 2`