diff --git a/taskcluster/tc-train-unittests.sh b/taskcluster/tc-train-unittests.sh
index 4ed6931c..4be52e90 100644
--- a/taskcluster/tc-train-unittests.sh
+++ b/taskcluster/tc-train-unittests.sh
@@ -4,6 +4,8 @@ set -xe
 
 source $(dirname "$0")/tc-tests-utils.sh
 
+extract_python_versions "$1" "pyver" "pyver_pkg" "py_unicode_type" "pyconf" "pyalias"
+
 mkdir -p ${TASKCLUSTER_ARTIFACTS} || true
 
 virtualenv_activate "${pyalias}" "deepspeech"