diff --git a/tenants/ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml b/tenants/ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml
index 5a668add..4c343e09 100644
--- a/tenants/ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml
+++ b/tenants/ai-example/dsp-example-pipeline/base/execute-kfp-task.yaml
@@ -24,5 +24,7 @@ spec:
       env:
         - name: KUBEFLOW_ENDPOINT
           value: $(inputs.params.KUBEFLOW_ENDPOINT)
+        - name: BASE_IMAGE
+          value: $(inputs.params.IMAGE):$(inputs.params.TAG)
       script: |
         python $(inputs.params.SCRIPT)