diff --git a/tests/test_demos.sh b/tests/test_demos.sh
index db63f194..3a54dd14 100644
--- a/tests/test_demos.sh
+++ b/tests/test_demos.sh
@@ -1,19 +1,17 @@
 #!/bin/bash
 
 cd ivy
-python3 -m pip install .
-conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
-
-cd ..
-cd demos
+pip install -e .
 
+cd ../demos
 mkdir .ivy
 touch .ivy/key.pem
 echo -n "$1" > .ivy/key.pem
 
+conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
+
 # install dependencies
-python3 -m pip install -r requirements.txt >/dev/null 2>&1
+pip install -r requirements.txt >/dev/null 2>&1
 
 # run test
-echo "$2"
-python3 tests/main.py "$2"
+python3 new_tests/main.py "$2"
\ No newline at end of file