diff --git a/pyproject.toml b/pyproject.toml
index f3791975..6415e307 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -71,7 +71,8 @@ qdrant-client = "*"
 vllm = "*"
 pytube = "*"
 pydub = "*"
-arize-mlflow = "*"
+arize = "*"
+mlflow = "*"
 sentence-transformers = "*"
 
 
diff --git a/requirements.txt b/requirements.txt
index 34992a67..5d7d71ff 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -75,5 +75,6 @@ qdrant-client
 vllm
 pytube
 pydub
-arize-mlflow
+arize
+mlflow
 sentence-transformers