diff --git a/setup.py b/setup.py
index 116aede62a..4c7bfdd733 100644
--- a/setup.py
+++ b/setup.py
@@ -47,7 +47,8 @@
 ]
 
 install_requires = [
-    'mosaicml[nlp,streaming,wandb]>=0.14.1,<0.15',
+    'composer[nlp,wandb]>=0.14.1,<0.15',
+    'mosaicml-streaming>=0.4.1,<0.5',
     'torch==1.13.1',
     'datasets==2.10.1',
     'sentencepiece==0.1.97',