diff --git a/transforms/universal/doc_id/spark/Dockerfile b/transforms/universal/doc_id/spark/Dockerfile
index 861146cab1..7176a9c304 100644
--- a/transforms/universal/doc_id/spark/Dockerfile
+++ b/transforms/universal/doc_id/spark/Dockerfile
@@ -1,6 +1,5 @@
-#ARG BASE_IMAGE=quay.io/dataprep1/data-prep-kit/data-prep-kit-spark-3.5.2:latest
-ARG BASE_IMAGE=data-prep-kit-spark-3.5.2:latest
-FROM ${BASE_IMAGE}
+ARG SPARK_BASE_IMAGE=data-prep-kit-spark-3.5.2:latest
+FROM ${SPARK_BASE_IMAGE}
 
 USER root
 # install pytest
diff --git a/transforms/universal/filter/spark/Dockerfile b/transforms/universal/filter/spark/Dockerfile
index f6dea8858c..12ebb4b95f 100644
--- a/transforms/universal/filter/spark/Dockerfile
+++ b/transforms/universal/filter/spark/Dockerfile
@@ -1,6 +1,5 @@
-ARG BASE_IMAGE=quay.io/dataprep1/data-prep-kit/data-prep-kit-spark-3.5.2:latest
-ARG BASE_IMAGE=data-prep-kit-spark-3.5.2:0.2.1
-FROM ${BASE_IMAGE}
+ARG SPARK_BASE_IMAGE=data-prep-kit-spark-3.5.2:latest
+FROM ${SPARK_BASE_IMAGE}
 
 USER root
 # install pytest
diff --git a/transforms/universal/noop/spark/Dockerfile b/transforms/universal/noop/spark/Dockerfile
index deb54f9740..94ba896a84 100644
--- a/transforms/universal/noop/spark/Dockerfile
+++ b/transforms/universal/noop/spark/Dockerfile
@@ -1,5 +1,5 @@
-ARG BASE_IMAGE=quay.io/dataprep1/data-prep-kit/data-prep-kit-spark-3.5.2:latest
-FROM ${BASE_IMAGE}
+ARG SPARK_BASE_IMAGE=data-prep-kit-spark-3.5.2:latest
+FROM ${SPARK_BASE_IMAGE}
 
 USER root
 # install pytest