diff --git a/test-mlflow-tracking/test-mlflow-tracking.ipynb b/test-mlflow-tracking/test-mlflow-tracking.ipynb
index a5ef639..5bf7e35 100644
--- a/test-mlflow-tracking/test-mlflow-tracking.ipynb
+++ b/test-mlflow-tracking/test-mlflow-tracking.ipynb
@@ -10,20 +10,7 @@
    },
    "outputs": [],
    "source": [
-    "! pip install mlflow"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c5d35479-7453-4ec9-84ee-686d7f779561",
-   "metadata": {
-    "scrolled": true,
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "! pip install xgboost"
+    "!pip install mlflow xgboost mlrun[complete]"
    ]
   },
   {