diff --git a/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py b/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py
index ea65f998c8..3b59cae321 100644
--- a/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py
+++ b/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py
@@ -64,6 +64,7 @@ def test_js_model_with_optimize_speculative_decoding_config_gated_requests_are_e
                 "Image": ANY,
                 "Environment": {
                     "SAGEMAKER_PROGRAM": "inference.py",
+                    "SAGEMAKER_MODEL_SERVER_TIMEOUT": "3600",
                     "ENDPOINT_SERVER_TIMEOUT": "3600",
                     "MODEL_CACHE_ROOT": "/opt/ml/model",
                     "SAGEMAKER_ENV": "1",
@@ -150,6 +151,7 @@ def test_js_model_with_optimize_sharding_and_resource_requirements_requests_are_
                 "Image": ANY,
                 "Environment": {
                     "SAGEMAKER_PROGRAM": "inference.py",
+                    "SAGEMAKER_MODEL_SERVER_TIMEOUT": "3600",
                     "ENDPOINT_SERVER_TIMEOUT": "3600",
                     "MODEL_CACHE_ROOT": "/opt/ml/model",
                     "SAGEMAKER_ENV": "1",
@@ -237,6 +239,7 @@ def test_js_model_with_optimize_quantization_on_pre_optimized_model_requests_are
                 "Image": ANY,
                 "Environment": {
                     "SAGEMAKER_PROGRAM": "inference.py",
+                    "SAGEMAKER_MODEL_SERVER_TIMEOUT": "3600",
                     "ENDPOINT_SERVER_TIMEOUT": "3600",
                     "MODEL_CACHE_ROOT": "/opt/ml/model",
                     "SAGEMAKER_ENV": "1",