diff --git a/setup.py b/setup.py
index e72139b..0352a77 100644
--- a/setup.py
+++ b/setup.py
@@ -64,7 +64,7 @@ def read(fname):
 
     # We don't declare our dependency on mxnet here because we build with
     # different packages for different variants (e.g. mxnet-mkl and mxnet-cu90).
-    install_requires=['sagemaker-training>=3.5.2', 'retrying==1.3.3'],
+    install_requires=['sagemaker-training>=3.5.2,<=4.8.3', 'retrying==1.3.3'],
     extras_require={
         'test': test_dependencies
     },