diff --git a/action.yml b/action.yml
index e797d41..4b92952 100644
--- a/action.yml
+++ b/action.yml
@@ -91,7 +91,7 @@ runs:
       if: ${{ inputs.workspace != '' }}
       with:
         source: .terraform
-        bucket: image-pipeline-assets
+        bucket: ${{ inputs.cache_bucket }}
         archive: true
         key: ${{ github.repository }}/${{ inputs.workspace }}/${{ steps.uuid.outputs.uuid }}/terraform-dir
         
@@ -100,7 +100,7 @@ runs:
       if: ${{ inputs.workspace == '' }}
       with:
         source: .terraform
-        bucket: image-pipeline-assets
+        bucket: ${{ inputs.cache_bucket }}
         archive: true
         key: ${{ github.repository }}/${{ steps.uuid.outputs.uuid  }}/terraform-dir
     
@@ -109,7 +109,7 @@ runs:
       if: ${{ inputs.workspace != '' }}
       with:
         source: .terraform.lock.hcl
-        bucket: image-pipeline-assets
+        bucket: ${{ inputs.cache_bucket }}
         archive: false
         key: ${{ github.repository }}/${{ inputs.workspace }}/${{ steps.uuid.outputs.uuid }}/terraform.lock.hcl
       
@@ -118,7 +118,7 @@ runs:
       if: ${{ inputs.workspace == '' }}
       with:
         source: .terraform.lock.hcl
-        bucket: image-pipeline-assets
+        bucket: ${{ inputs.cache_bucket }}
         archive: false
         key: ${{ github.repository }}/${{ steps.uuid.outputs.uuid }}/terraform.lock.hcl