diff --git a/cdk.json b/cdk.json
index c701c62..8bf33de 100644
--- a/cdk.json
+++ b/cdk.json
@@ -65,7 +65,7 @@
     },
 
     "prod": {
-      "IMAGE_PATH_AND_TAG": "ghcr.io/sage-bionetworks/schematic:v23.12.1",
+      "IMAGE_PATH_AND_TAG": "ghcr.io/sage-bionetworks/schematic:v24.1.1",
       "AWS_DEFAULT_REGION": "us-east-1",
       "PORT": "443",
       "TAGS": {
diff --git a/docker_fargate/docker_fargate_stack.py b/docker_fargate/docker_fargate_stack.py
index a040380..3d4ffb6 100644
--- a/docker_fargate/docker_fargate_stack.py
+++ b/docker_fargate/docker_fargate_stack.py
@@ -90,6 +90,7 @@ def __init__(self, scope: Construct, context: str, env: dict, vpc: ec2.Vpc, **kw
             task_image_options=task_image_options,
             memory_limit_mib=8192,      # Default is 512; 8192 MiB is equivalent to 8GB.
             public_load_balancer=True,  # Default is False
+            idle_timeout=Duration.seconds(300), # Modify default idle time out to avoid 504 gateway error
             # TLS:
             certificate=cert,
             protocol=elbv2.ApplicationProtocol.HTTPS,