diff --git a/api_endpoint/endpoint_methods/variables.tf b/api_endpoint/endpoint_methods/variables.tf
index 0b1013f..f6d7556 100644
--- a/api_endpoint/endpoint_methods/variables.tf
+++ b/api_endpoint/endpoint_methods/variables.tf
@@ -43,7 +43,6 @@ variable "reserved_concurrent_executions" {
   description = "The number of simultaneous executions to reserve for the function."
   default     = -1
 }
-
 variable "memory_size" {
   type        = number
   description = "The amount of memory, in MB, your Lambda Function is given. Valid values are from 128 to 10,240. Default is 128. 1,769 is equivalent to 1 vCPU."
diff --git a/main.tf b/main.tf
index 9501d75..2488e31 100644
--- a/main.tf
+++ b/main.tf
@@ -96,10 +96,6 @@ resource "aws_api_gateway_stage" "api_gateway_stage" {
   rest_api_id   = aws_api_gateway_rest_api.api_gateway.id
   deployment_id = aws_api_gateway_deployment.api_gateway_deployment.id
   stage_name    = "${var.app_name}-stage"
-
-  lifecycle {
-    create_before_destroy = true
-  }
 }
 
 # ========== Custom API URL ==========
diff --git a/variables.tf b/variables.tf
index 442dce4..801c537 100644
--- a/variables.tf
+++ b/variables.tf
@@ -46,6 +46,7 @@ variable "lambda_endpoint_definitions" {
       command                        = list(string)
       timeout                        = optional(number)
       reserved_concurrent_executions = optional(number)
+      memory_size                    = optional(number)
     }))
   }))
   description = "The definitions for each lambda function."