diff --git a/pyproject.toml b/pyproject.toml
index cb84e69d72..b6fbec18b8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
     "torch>=2.2.0",
     "lightning==2.3.0.dev20240328",
     "jsonargparse[signatures]>=4.27.6",
-    "litserve",   # imported by litgpt.deploy
+    "litserve==0.0.0.dev2",   # imported by litgpt.deploy
 ]
 
 [project.urls]