diff --git a/buildspec.yml b/buildspec.yml
index dfa02d60..faa2a490 100644
--- a/buildspec.yml
+++ b/buildspec.yml
@@ -74,7 +74,7 @@ phases:
       - aws s3 cp --region us-east-1 ../workshop/comfy.yaml s3://aws-gcr-solutions-us-east-1/extension-for-stable-diffusion-on-aws/comfy.yaml
       - aws s3api put-object-acl --region us-east-1 --bucket "aws-gcr-solutions-us-east-1" --key "extension-for-stable-diffusion-on-aws/comfy.yaml" --acl public-read
       - echo ""
-      - sed -i "s/{{latest}}/$TAG_NUMBER/" ../workshop/comfy_new.yaml
+      - sed -i "s/{{latest}}/$BUILD_VERSION/" ../workshop/comfy_new.yaml
       - aws s3 cp --region us-east-1 ../workshop/comfy_new.yaml s3://aws-gcr-solutions-us-east-1/extension-for-stable-diffusion-on-aws/comfy_dev.yaml
       - aws s3api put-object-acl --region us-east-1 --bucket "aws-gcr-solutions-us-east-1" --key "extension-for-stable-diffusion-on-aws/comfy_dev.yaml" --acl public-read
       - echo ""