diff --git a/tests/notebook_testing.py b/tests/notebook_testing.py
index 25f77d32..c07b710b 100644
--- a/tests/notebook_testing.py
+++ b/tests/notebook_testing.py
@@ -176,7 +176,7 @@ def test_notebook(nb):
     # kc.execute("ivy.set_backend('jax')")
     # kc.execute("ivy.unset_backend()")
     # TODO
-    kc.execute("import os;os.environ['IVY_ROOT']='./.ivy'")
+    # kc.execute("import os;os.environ['IVY_ROOT']='./.ivy'")
 
     while True:
         try:
diff --git a/tests/test_demos.sh b/tests/test_demos.sh
index f0aa2c31..7744e494 100755
--- a/tests/test_demos.sh
+++ b/tests/test_demos.sh
@@ -9,6 +9,7 @@ cd demos
 mkdir .ivy
 touch .ivy/key.pem
 echo -n "$1"> .ivy/key.pem
+export IVY_ROOT='./.ivy'
 
 # install dependencies
 python3 -m pip install -r requirements.txt >/dev/null 2>&1