diff --git a/main.tf b/main.tf
index 9b2aaee..abe44a2 100644
--- a/main.tf
+++ b/main.tf
@@ -97,6 +97,10 @@ module "metaflow-computation" {
   launch_template_http_endpoint               = var.launch_template_http_endpoint
   launch_template_http_tokens                 = var.launch_template_http_tokens
   launch_template_http_put_response_hop_limit = var.launch_template_http_put_response_hop_limit
+  job_state_time_limit_action                 = var.job_state_time_limit_action
+  job_state_time_limit_timeout                = var.job_state_time_limit_timeout
+  job_state_time_limit_reason                 = var.job_state_time_limit_reason
+
 
   standard_tags = var.tags
 }
diff --git a/modules/computation/variables.tf b/modules/computation/variables.tf
index 92d11ba..4ae5df5 100644
--- a/modules/computation/variables.tf
+++ b/modules/computation/variables.tf
@@ -102,3 +102,18 @@ variable "launch_template_image_id" {
   nullable    = true
   default     = null
 }
+
+variable "job_state_time_limit_action" {
+  type        = string
+  description = "The action to take when the job times out"
+}
+
+variable "job_state_time_limit_timeout" {
+  type        = number
+  description = "The time limit in seconds for the job to run before the action is taken"
+}
+
+variable "job_state_time_limit_reason" {
+  type        = number
+  description = "The reason for the job state time limit action"
+}
diff --git a/variables.tf b/variables.tf
index 1738c0b..0b0758e 100644
--- a/variables.tf
+++ b/variables.tf
@@ -199,3 +199,18 @@ variable "enable_key_rotation" {
   description = "Enable key rotation for KMS keys"
   default     = false
 }
+
+variable "job_state_time_limit_action" {
+  type        = string
+  description = "The action to take when the job times out"
+}
+
+variable "job_state_time_limit_timeout" {
+  type        = number
+  description = "The time limit in seconds for the job to run before the action is taken"
+}
+
+variable "job_state_time_limit_reason" {
+  type        = number
+  description = "The reason for the job state time limit action"
+}