diff --git a/setup.py b/setup.py
index 024b67a7..00c61249 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,8 @@
     "click>=7.1.2",
     "retry>=0.9.2",
     "requests>=2.24.0",
-    "mlflow>=1.23.0",
+    "mlflow==1.26.0",
+    "protobuf==3.20.1",
     "tqdm>=4.50.0",
     "azure-identity>=1.7.1",
     "azure-mgmt-datafactory>=2.2.0",