diff --git a/.github/workflows/terraform-drift.yml b/.github/workflows/terraform-drift.yml
index 5a3fe2a..fcc9a07 100644
--- a/.github/workflows/terraform-drift.yml
+++ b/.github/workflows/terraform-drift.yml
@@ -9,36 +9,68 @@ on:
         required: false
 
     inputs:
+      aws-account-id:
+        description: "The AWS account ID to deploy to"
+        required: true
+        type: string
+
+      aws-region:
+        default: "eu-west-2"
+        description: "The AWS region to deploy to"
+        required: false
+        type: string
+
+      environment:
+        default: "production"
+        description: "The environment to deploy to"
+        required: false
+        type: string
+
+      use-env-as-suffix:
+        default: false
+        description: "Whether to use the environment as a suffix for the state file and iam roles"
+        required: false
+        type: boolean
+
       runs-on:
         default: "ubuntu-latest"
         description: "Single label value for the GitHub runner to use (custom value only applies to Terraform Plan and Apply steps)"
         required: false
         type: string
 
-      aws-account-id:
-        description: "The AWS account ID to deploy to"
-        required: true
+      terraform-dir:
+        default: "."
+        description: "The directory to validate"
+        required: false
         type: string
 
-      aws-role:
-        default: "${{ github.event.repository.name }}"
-        description: "The AWS role to assume"
+      terraform-lock-timeout:
+        default: "30s"
+        description: The time to wait for a lock
         required: false
         type: string
 
-      aws-read-role-name:
-        description: "Overrides the default behavior, and uses a custom role name for read-only access"
+      terraform-state-key:
+        default: ""
+        description: "The key of the terraform state (default: <repo-name>.tfstate)"
         required: false
         type: string
 
-      aws-write-role-name:
-        description: "Overrides the default behavior, and uses a custom role name for read-write access"
+      terraform-values-file:
+        default: ""
+        description: "The values file to use (default: <environment>.tfvars)"
         required: false
         type: string
 
-      aws-region:
-        default: "eu-west-2"
-        description: "The AWS region to deploy to"
+      terraform-version:
+        default: "1.7.1"
+        description: "The version of terraform to use"
+        required: false
+        type: string
+
+      working-directory:
+        default: "."
+        description: "The working directory to run terraform commands in"
         required: false
         type: string