diff --git a/examples/prompt-tuning-sst2.ipynb b/examples/prompt-tuning-sst2.ipynb
index 05938b0..b841406 100644
--- a/examples/prompt-tuning-sst2.ipynb
+++ b/examples/prompt-tuning-sst2.ipynb
@@ -36,7 +36,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install -q petals datasets wandb"
+    "%pip install -q git+https://github.com/bigscience-workshop/petals",
+    "%pip install -q datasets wandb"
    ]
   },
   {