diff --git a/python/run-tests b/python/run-tests
index 80acd002ab7eb2eae397a41f38ab7543d95eb237..a4f0cac059ff3f21b2fbdf7dbcd8fb465e13b082 100755
--- a/python/run-tests
+++ b/python/run-tests
@@ -41,7 +41,7 @@ function run_test() {
 
     # Fail and exit on the first test failure.
     if [[ $FAILED != 0 ]]; then
-        cat unit-tests.log | grep -v "^[0-9][0-9]*" # filter all lines starting with a number.
+        cat $LOG_FILE | grep -v "^[0-9][0-9]*" # filter all lines starting with a number.
         echo -en "\033[31m"  # Red
         echo "Had test failures; see logs."
         echo -en "\033[0m"  # No color
@@ -87,7 +87,7 @@ function run_streaming_tests() {
     run_test "pyspark/streaming/tests.py"
 }
 
-echo "Running PySpark tests. Output is in python/unit-tests.log."
+echo "Running PySpark tests. Output is in python/$LOG_FILE."
 
 export PYSPARK_PYTHON="python"