diff --git a/setup.py b/setup.py
index 1694c90574..87f678f820 100644
--- a/setup.py
+++ b/setup.py
@@ -186,7 +186,7 @@ def package_files(prefix: str, directory: str, extension: str):
 ]
 
 extra_deps['streaming'] = [
-    'mosaicml-streaming<0.3',
+    'mosaicml-streaming<0.4',
     'boto3>=1.21.45,<2',
     'paramiko>=2.11.0,<3',
 ]