diff --git a/amplify.yml b/amplify.yml
index 4f0cfe2cc..1c3c52a21 100644
--- a/amplify.yml
+++ b/amplify.yml
@@ -6,8 +6,8 @@ frontend:
         - nvm install 18.13
         - nvm use 18.13
         - |
-          echo "${AWS_BRANCH_ARN#*/branches/}"
-          if [[ "${AWS_BRANCH}" != "main" ]] && [["${AWS_BRANCH}" != "staging"]]; then
+          echo "${AWS_BRANCH}"
+          if [[ "${AWS_BRANCH}" != "main" && "${AWS_BRANCH}" != "staging"]]; then
             PREVIEW_URL="https://${AWS_BRANCH}.${AWS_APP_ID}.amplifyapp.com";
             NEXTAUTH_URL=$PREVIEW_URL;
             echo "PREVIEW_URL=$PREVIEW_URL" >> .env;