diff --git a/pavelib/utils/test/utils.py b/pavelib/utils/test/utils.py
index 0a2a35ec04..0e5d08961d 100644
--- a/pavelib/utils/test/utils.py
+++ b/pavelib/utils/test/utils.py
@@ -23,7 +23,7 @@ MINIMUM_FIREFOX_VERSION = 28.0
 
 COVERAGE_CACHE_BUCKET = "edx-tools-coverage-caches"
 COVERAGE_CACHE_BASEPATH = "test_root/who_tests_what"
-COVERAGE_CACHE_BASELINE = "who_tests_what.{}.baseline".format(os.environ.get('TEST_SUITE', 'all'))
+COVERAGE_CACHE_BASELINE = "who_tests_what.{}.baseline".format(os.environ.get('WTW_CONTEXT', 'all'))
 WHO_TESTS_WHAT_DIFF = "who_tests_what.diff"
 
 
diff --git a/scripts/Jenkinsfiles/bokchoy b/scripts/Jenkinsfiles/bokchoy
index e73e67e20d..339f952b1d 100644
--- a/scripts/Jenkinsfiles/bokchoy
+++ b/scripts/Jenkinsfiles/bokchoy
@@ -33,6 +33,9 @@ pipeline {
         timestamps()
         timeout(60)
     }
+    environment {
+        WTW_CONTEXT = "bokchoy"
+    }
     stages {
         stage('Mark build as pending on Github') {
             when {
diff --git a/scripts/Jenkinsfiles/python b/scripts/Jenkinsfiles/python
index 5a42a4394e..beb4c6f482 100644
--- a/scripts/Jenkinsfiles/python
+++ b/scripts/Jenkinsfiles/python
@@ -52,6 +52,7 @@ pipeline {
         XDIST_WORKER_KEY_NAME = "jenkins-worker"
         XDIST_WORKER_SUBNET = credentials('XDIST_WORKER_SUBNET')
         XDIST_WORKER_SECURITY_GROUP = credentials('XDIST_WORKER_SECURITY_GROUP')
+        WTW_CONTEXT = "python"
     }
     stages {
         stage('Mark build as pending on Github') {