diff --git a/docs/notebooks/Data_structures.ipynb b/docs/notebooks/Data_structures.ipynb
index 1ad1e6abb..a3337186c 100644
--- a/docs/notebooks/Data_structures.ipynb
+++ b/docs/notebooks/Data_structures.ipynb
@@ -56,7 +56,7 @@
       "source": [
         "# This should take care of all the dependencies on colab:\n",
         "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-        "!pip install -qqq sleap[pypi]\n",
+        "!pip install -qqq sleap==1.3.1111\n",
         "\n",
         "# But to do it locally, we'd recommend the conda package (available on Windows + Linux):\n",
         "# conda create -n sleap -c sleap -c conda-forge -c nvidia sleap"
diff --git a/docs/notebooks/Interactive_and_realtime_inference.ipynb b/docs/notebooks/Interactive_and_realtime_inference.ipynb
index 8d0107fa7..5d9c7e33d 100644
--- a/docs/notebooks/Interactive_and_realtime_inference.ipynb
+++ b/docs/notebooks/Interactive_and_realtime_inference.ipynb
@@ -60,7 +60,7 @@
       "source": [
         "# This should take care of all the dependencies on colab:\n",
         "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-        "!pip install -qqq sleap[pypi]\n",
+        "!pip install -qqq sleap==1.3.1\n",
         "\n",
         "\n",
         "# But to do it locally, we'd recommend the conda package (available on Windows + Linux):\n",
diff --git a/docs/notebooks/Interactive_and_resumable_training.ipynb b/docs/notebooks/Interactive_and_resumable_training.ipynb
index 708d10845..be82c19a5 100644
--- a/docs/notebooks/Interactive_and_resumable_training.ipynb
+++ b/docs/notebooks/Interactive_and_resumable_training.ipynb
@@ -62,7 +62,7 @@
       "source": [
         "# This should take care of all the dependencies on colab:\n",
         "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-        "!pip install -qqq sleap[pypi]\n",
+        "!pip install -qqq sleap==1.3.1\n",
         "\n",
         "\n",
         "# But to do it locally, we'd recommend the conda package (available on Windows + Linux):\n",
diff --git a/docs/notebooks/Model_evaluation.ipynb b/docs/notebooks/Model_evaluation.ipynb
index 9bc55953d..62bf3935a 100644
--- a/docs/notebooks/Model_evaluation.ipynb
+++ b/docs/notebooks/Model_evaluation.ipynb
@@ -40,7 +40,7 @@
       ],
       "source": [
         "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-        "!pip install -qqq sleap[pypi]\n",
+        "!pip install -qqq sleap==1.3.1\n",
         "!apt -qq install tree\n",
         "!wget -q https://storage.googleapis.com/sleap-data/reference/flies13/td_fast.210505_012601.centered_instance.n%3D1800.zip\n",
         "!unzip -qq -o -d \"td_fast.210505_012601.centered_instance.n=1800\" \"td_fast.210505_012601.centered_instance.n=1800.zip\""
diff --git a/docs/notebooks/Post_inference_tracking.ipynb b/docs/notebooks/Post_inference_tracking.ipynb
index cfd73c99f..106c7ae88 100644
--- a/docs/notebooks/Post_inference_tracking.ipynb
+++ b/docs/notebooks/Post_inference_tracking.ipynb
@@ -61,7 +61,7 @@
       "source": [
         "# This should take care of all the dependencies on colab:\n",
         "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-        "!pip install -qqq sleap[pypi]\n",
+        "!pip install -qqq sleap==1.3.1\n",
         "\n",
         "# But to do it locally, we'd recommend the conda package (available on Windows + Linux):\n",
         "# conda create -n sleap -c sleap -c conda-forge -c nvidia sleap"
diff --git a/docs/notebooks/Training_and_inference_on_an_example_dataset.ipynb b/docs/notebooks/Training_and_inference_on_an_example_dataset.ipynb
index b5d2fa78d..22c4193f7 100644
--- a/docs/notebooks/Training_and_inference_on_an_example_dataset.ipynb
+++ b/docs/notebooks/Training_and_inference_on_an_example_dataset.ipynb
@@ -62,7 +62,7 @@
    ],
    "source": [
     "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-    "!pip install -qqq sleap[pypi]"
+    "!pip install -qqq sleap==1.3.1"
    ]
   },
   {
diff --git a/docs/notebooks/Training_and_inference_using_Google_Drive.ipynb b/docs/notebooks/Training_and_inference_using_Google_Drive.ipynb
index 0a3fc505b..96374982d 100644
--- a/docs/notebooks/Training_and_inference_using_Google_Drive.ipynb
+++ b/docs/notebooks/Training_and_inference_using_Google_Drive.ipynb
@@ -59,7 +59,7 @@
    ],
    "source": [
     "!pip uninstall -qqq -y opencv-python opencv-contrib-python\n",
-    "!pip install -qqq sleap[pypi]"
+    "!pip install -qqq sleap==1.3.1"
    ]
   },
   {