diff --git a/python/run-tests.py b/python/run-tests.py
index f5857f8c6221407772b759d2f282e50277015cdd..ee73eb1506ca42951ca18ee18bae4b43ad6bee6a 100755
--- a/python/run-tests.py
+++ b/python/run-tests.py
@@ -56,7 +56,8 @@ LOGGER = logging.getLogger()
 
 def run_individual_python_test(test_name, pyspark_python):
     env = dict(os.environ)
-    env.update({'SPARK_TESTING': '1', 'PYSPARK_PYTHON': which(pyspark_python)})
+    env.update({'SPARK_TESTING': '1', 'PYSPARK_PYTHON': which(pyspark_python),
+                'PYSPARK_DRIVER_PYTHON': which(pyspark_python)})
     LOGGER.debug("Starting test(%s): %s", pyspark_python, test_name)
     start_time = time.time()
     try: