diff --git a/workflows/mlflow-e2e.yaml b/workflows/mlflow-e2e.yaml
index 056b02c..e769438 100644
--- a/workflows/mlflow-e2e.yaml
+++ b/workflows/mlflow-e2e.yaml
@@ -17,7 +17,7 @@ outputs:
     type: string
 steps:
   - name: builder
-    image: ghcr.io/fuseml/mlflow-builder:dev
+    image: ghcr.io/fuseml/mlflow-builder:v0.2
     inputs:
       - name: mlflow-codeset
         codeset:
@@ -42,7 +42,7 @@ steps:
         product: mlflow
         service_resource: s3
   - name: predictor
-    image: ghcr.io/fuseml/kfserving-predictor:dev
+    image: ghcr.io/fuseml/kfserving-predictor:0.2
     inputs:
       - name: model
         value: '{{ steps.trainer.outputs.mlflow-model-url }}'