diff --git a/buildspec.yml b/buildspec.yml
index e2c1e81..42e777e 100644
--- a/buildspec.yml
+++ b/buildspec.yml
@@ -16,7 +16,7 @@ phases:
 
   build:
     commands:
-      - if [ -z $VERSION ]; then export VERSION=$CODEBUILD_SOURCE_VERSION; fi
+      - if [ -z $VERSION ]; then export VERSION=$(echo "$CODEBUILD_WEBHOOK_TRIGGER" | cut -f 2 -d'/'); fi
       - sed -i "s|<VERSION>|$VERSION|g" template.yaml
       - aws s3 cp $DEPLOYMENT_ASSET_LOCATION sf-deploy.py #Mandatory
       - python sf-deploy.py # Mandatory