diff --git a/tox.ini b/tox.ini
index aedfeed9..3ee46f4e 100644
--- a/tox.ini
+++ b/tox.ini
@@ -62,7 +62,7 @@ deps =
     mock
     requests == 2.20.0
     urllib3 < 1.23, >= 1.21
-    sagemaker
+    sagemaker < 2
     sagemaker-containers
     torch
     torchvision