diff --git a/code/run_index.sh b/code/run_index.sh
index 12f644c..9f37710 100644
--- a/code/run_index.sh
+++ b/code/run_index.sh
@@ -9,6 +9,7 @@ OPENAI_API_KEY="${OPENAI_API_KEY:-}"
 : "${OPENAI_API_KEY:?Set OPENAI_API_KEY before running run_index.sh}"
 export OPENAI_API_KEY
 export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"
+PYTHON="${PYTHON:-python3}"
 
 # Input and model parameters.
 CORPUS="/home/wcj/data/project/memgraphrag/input/smallcorpus/smallcorpus.txt"
diff --git a/code/run_retrieval_test.sh b/code/run_retrieval_test.sh
index ffe5c54..47ea9d8 100755
--- a/code/run_retrieval_test.sh
+++ b/code/run_retrieval_test.sh
@@ -6,6 +6,7 @@ SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 
 # 使用的 GPU 编号
 export CUDA_VISIBLE_DEVICES="0"
+PYTHON="${PYTHON:-python3}"
 
 # OpenAI 兼容接口的 API Key；使用无鉴权的本地服务时可留空
 export OPENAI_API_KEY=