diff --git a/components/aws/sagemaker/tests/unit_tests/requirements.txt b/components/aws/sagemaker/tests/unit_tests/requirements.txt
index cb309fa6..028a10c6 100644
--- a/components/aws/sagemaker/tests/unit_tests/requirements.txt
+++ b/components/aws/sagemaker/tests/unit_tests/requirements.txt
@@ -3,4 +3,4 @@ coverage==5.1
 pathlib2==2.3.5
 pytest==5.4.1
 pyyaml==5.3.1
-sagemaker==1.56.1
\ No newline at end of file
+sagemaker==2.128.0
\ No newline at end of file