diff --git a/run_ortmodule_mvp.sh b/run_ortmodule_mvp.sh
index ae5e3f5eb3..5bb742eeee 100755
--- a/run_ortmodule_mvp.sh
+++ b/run_ortmodule_mvp.sh
@@ -4,12 +4,12 @@ cur_dir=$(basename `pwd`)
 
 if [[ ${cur_dir} != "RelWithDebInfo" ]]
 then
-    echo "Going to build folder"
+    echo "Going to build folder (aka build/Linux/RelWithDebInfo)"
     cd build/Linux/RelWithDebInfo
 fi
 
 echo "Exporting PYTHONPATH to use build dir as onnxruntime package"
-export PYTHONPATH=/home/thiagofc/dev/github/onnxruntime/build/Linux/RelWithDebInfo/
+export PYTHONPATH=$(pwd)
 
 echo "Copying PyTorch frontend source-code to build folder"
 cp -Rf ../../../orttraining/orttraining/python/training/* ../../../build/Linux/RelWithDebInfo/onnxruntime/training/