diff --git a/pkgs/development/python-modules/mlflow/default.nix b/pkgs/development/python-modules/mlflow/default.nix
index 7032d0e91d51..6646fad25e16 100644
--- a/pkgs/development/python-modules/mlflow/default.nix
+++ b/pkgs/development/python-modules/mlflow/default.nix
@@ -57,7 +57,9 @@ buildPythonPackage rec {
   pythonRemoveDeps = [ "shap" ];
   pythonRelaxDeps = [
     "gunicorn"
+    "importlib-metadata"
     "packaging"
+    "protobuf"
     "pytz"
     "pyarrow"
   ];