diff --git a/pavelib/utils/test/utils.py b/pavelib/utils/test/utils.py
index 0a2a35ec0429765a44e3b11140030cd83c68a5a1..0e5d08961d5bdd2864bad7ea3d129d96e9331045 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 547c01b2d977ddc5e1020375ba372e5892fe71f4..988e7adfe58923a52e9abe13f8230b2cc510dbe1 100644
--- a/scripts/Jenkinsfiles/bokchoy
+++ b/scripts/Jenkinsfiles/bokchoy
@@ -31,6 +31,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 cb9414fc129c2c70d00be81fc19c572fa9976249..604462f4cd54db75b74ad9404b3485323faa2681 100644
--- a/scripts/Jenkinsfiles/python
+++ b/scripts/Jenkinsfiles/python
@@ -50,6 +50,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') {