diff --git a/sdk/python/feast/pyspark/launchers/k8s/k8s_utils.py b/sdk/python/feast/pyspark/launchers/k8s/k8s_utils.py
index bf03f149ff..5f57f19029 100644
--- a/sdk/python/feast/pyspark/launchers/k8s/k8s_utils.py
+++ b/sdk/python/feast/pyspark/launchers/k8s/k8s_utils.py
@@ -256,7 +256,7 @@ def _cancel_job_by_id(api: CustomObjectsApi, namespace: str, job_id: str):
 spec:
   type: Scala
   mode: cluster
-  image: "gcr.io/kf-feast/spark-py-extra:v3.0.1"
+  image: "gcr.io/kf-feast/spark-py:v3.0.1"
   imagePullPolicy: Always
   sparkVersion: "3.0.1"
   timeToLiveSeconds: 3600