diff --git a/modules/identity-token-rotation-lambda/main.tf b/modules/identity-token-rotation-lambda/main.tf
index a5804f3..1dd69de 100644
--- a/modules/identity-token-rotation-lambda/main.tf
+++ b/modules/identity-token-rotation-lambda/main.tf
@@ -17,6 +17,9 @@ module "lambda_layer" {
   compatible_runtimes      = [var.lambda_runtime]
   runtime                  = var.lambda_runtime
   recreate_missing_package = false
+  build_in_docker          = var.build_in_docker
+//  docker_image             = "public.ecr.aws/lambda/python:3.8"
+
   source_path = [
     {
       path             = path.module,
diff --git a/modules/identity-token-rotation-lambda/requirements.txt b/modules/identity-token-rotation-lambda/requirements.txt
index 54b6c70..15b087b 100644
--- a/modules/identity-token-rotation-lambda/requirements.txt
+++ b/modules/identity-token-rotation-lambda/requirements.txt
@@ -1,4 +1,4 @@
-arrow==1.3.0
+# arrow==1.2.3
 boto3==1.34.71
 aws-lambda-powertools[tracer]==2.35.1
 requests==2.31.0
diff --git a/modules/identity-token-rotation-lambda/variable.tf b/modules/identity-token-rotation-lambda/variable.tf
index 63ec863..d7ac326 100644
--- a/modules/identity-token-rotation-lambda/variable.tf
+++ b/modules/identity-token-rotation-lambda/variable.tf
@@ -23,4 +23,8 @@ variable "use_janus_proxy" {
 variable "rotation_schedule_expression" {
   type = string
   default = "rate(4 hours)"
-}
\ No newline at end of file
+}
+variable "build_in_docker" {
+  type = bool
+  default = false
+}