diff --git a/action.yml b/action.yml
index 57a887d..10be0e0 100644
--- a/action.yml
+++ b/action.yml
@@ -17,24 +17,23 @@ inputs:
     required: false
     default: false
   arch:
-    description: Architecture of the linux host
+    description: Allowed values are - amd64, arm64
     required: false
-    default: x86
+    default: amd64
 outputs:
   version:
     description: "The AWS CLI version that was installed"
     value: ${{ steps.set-output.outputs.version }}
 
-
 runs:
   using: "composite"
   steps:
     - id: set-env-vars
       run: |
         echo "AWS_CLI_VERSION=${{ inputs.version }}" >> $GITHUB_ENV
+        echo "AWS_CLI_ARCH=${{ inputs.arch }}" >> $GITHUB_ENV        
         echo "VERBOSE=${{ inputs.verbose }}" >> $GITHUB_ENV
         echo "LIGHTSAILCTL=${{ inputs.lightsailctl }}" >> $GITHUB_ENV
-        echo "ARCH=${{ inputs.arch }}" >> $GITHUB_ENV
       shell: bash
     - id: install-aws-cli
       run: sudo --preserve-env ${{ github.action_path }}/entrypoint.sh