2 files changed, 3 insertions, 2 deletions
diff --git a/benchmark-terasort b/benchmark-terasort
index 1eb6b74..0105571 100755
--- a/benchmark-terasort
+++ b/benchmark-terasort
@@ -22,8 +22,8 @@ THIS_BENCHMARK_DIR="$(cd $(dirname "${BASH_SOURCE[0]}") && pwd -P)"
 
 source $THIS_BENCHMARK_DIR/common.sh
 
-if [ ! -d $THIS_BENCHMARK_DIR/${NGIGABYTES}GB ]; then
-    echo "error: ${NGIGABYTES}GB baseline directory is missing!"
+if [ ! -d $TERAGEN_BASELINE_DIR/${NGIGABYTES}GB ]; then
+    echo "error: ${NGIGABYTES}GB baseline file is missing!"
     exit 1
 fi
 
diff --git a/common.sh b/common.sh
index 99dfd88..a6fa7ae 100644
--- a/common.sh
+++ b/common.sh
@@ -25,3 +25,4 @@
 : ${USER:=$LOGNAME}
 : ${RESULTS_FILE=$BENCHMARK_RESULTS_DIR/results.csv}
 : ${RESULTS_DB=$BENCHMARK_RESULTS_DIR/results.db}
+: ${TERAGEN_BASELINE_DIR:=$HOME/hadoop-terasort-reference-files}