diff --git a/.make.versions b/.make.versions
index 0bc1e8c33..a8b443a98 100644
--- a/.make.versions
+++ b/.make.versions
@@ -40,6 +40,7 @@ LANG_ID_PYTHON_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 LANG_ID_RAY_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 
 TOKENIZATION_RAY_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
+TOKENIZATION_PYTHON_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 
 MALWARE_RAY_VERSION=0.5.0$(RELEASE_VERSION_SUFFIX)
 
@@ -47,9 +48,6 @@ PROGLANG_SELECT_RAY_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 
 CODE_QUALITY_RAY_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 
-DOC_QUALITY_PYTHON_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
-DOC_QUALITY_RAY_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
-
 INGEST_TO_PARQUET_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 INGEST_TO_PARQUET_RAY_VERSION=0.4.0$(RELEASE_VERSION_SUFFIX)
 
diff --git a/transforms/universal/tokenization/python/Makefile b/transforms/universal/tokenization/python/Makefile
index 9b11d8cdd..5a3851821 100644
--- a/transforms/universal/tokenization/python/Makefile
+++ b/transforms/universal/tokenization/python/Makefile
@@ -8,7 +8,7 @@ include $(REPOROOT)/transforms/.make.transforms
 
 TRANSFORM_NAME=tokenization
 # $(REPOROOT)/.make.versions file contains the versions
-DOCKER_IMAGE_VERSION=${TOKENIZATION_VERSION}
+DOCKER_IMAGE_VERSION=${TOKENIZATION_PYTHON_VERSION}
 
 venv::	.transforms.python-venv