diff --git a/docs/integrations/telemetry/mlflow-tracing.ipynb b/docs/integrations/telemetry/mlflow-tracing.ipynb
index 27ec3cb16..53698dba6 100644
--- a/docs/integrations/telemetry/mlflow-tracing.ipynb
+++ b/docs/integrations/telemetry/mlflow-tracing.ipynb
@@ -36,8 +36,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# ! pip install \"guardrails-ai[databricks]\" -q\n",
-    "! pip install \"git+https://github.com/guardrails-ai/guardrails.git@mlflow-integration\" mlflow -q"
+    "! pip install \"guardrails-ai[databricks]\" -q\n"
    ]
   },
   {