diff --git a/pyproject.toml b/pyproject.toml
index b667933076844ec6c023d6b513f212ce0cb6c4a3..ab6b90f045455dea08af4610340fa5ecf1c35ca3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -53,7 +53,7 @@ dependencies = [
   "grad-cam~=1.5.0",
   "versioningit",
   "python-gitlab",
-  "mlflow",
+  "mlflow-skinny",
 ]
 
 [project.urls]
@@ -109,7 +109,7 @@ tensorboard = "*"
 grad-cam = "~=1.5.0"
 versioningit = "*"
 python-gitlab = "*"
-mlflow = "*"
+mlflow-skinny = "*"
 
 [tool.pixi.feature.self.pypi-dependencies]
 mednet = { path = ".", editable = true }