diff --git a/3_distributed_training/distributed_training_sm_unified_studio/scripts/requirements.txt b/3_distributed_training/distributed_training_sm_unified_studio/scripts/requirements.txt
index 1a65f9c2..25d97def 100644
--- a/3_distributed_training/distributed_training_sm_unified_studio/scripts/requirements.txt
+++ b/3_distributed_training/distributed_training_sm_unified_studio/scripts/requirements.txt
@@ -7,7 +7,7 @@ evaluate==0.4.3
 huggingface_hub[hf_transfer]
 mlflow
 safetensors>=0.5.2
-sagemaker==2.244.0
+sagemaker==3.8.0
 sagemaker-mlflow==0.1.0
 sentencepiece==0.2.0
 scikit-learn==1.6.1