diff --git a/examples/16_Train_a_text_labeler.ipynb b/examples/16_Train_a_text_labeler.ipynb
index 6eab347b..ccdcc8b7 100644
--- a/examples/16_Train_a_text_labeler.ipynb
+++ b/examples/16_Train_a_text_labeler.ipynb
@@ -52,7 +52,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets pandas"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets pandas"
       ],
       "execution_count": null,
       "outputs": []
diff --git a/examples/17_Train_without_labels.ipynb b/examples/17_Train_without_labels.ipynb
index bcb36636..f1318daf 100644
--- a/examples/17_Train_without_labels.ipynb
+++ b/examples/17_Train_without_labels.ipynb
@@ -45,7 +45,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets pandas"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets pandas"
       ],
       "execution_count": null,
       "outputs": []
diff --git a/examples/19_Train_a_QA_model.ipynb b/examples/19_Train_a_QA_model.ipynb
index 68454f0a..23f0d89e 100644
--- a/examples/19_Train_a_QA_model.ipynb
+++ b/examples/19_Train_a_QA_model.ipynb
@@ -47,7 +47,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets pandas"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets pandas"
       ],
       "execution_count": null,
       "outputs": []
@@ -292,4 +292,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/examples/20_Extractive_QA_to_build_structured_data.ipynb b/examples/20_Extractive_QA_to_build_structured_data.ipynb
index 5328dde1..e5308131 100644
--- a/examples/20_Extractive_QA_to_build_structured_data.ipynb
+++ b/examples/20_Extractive_QA_to_build_structured_data.ipynb
@@ -42,7 +42,7 @@
       },
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train]"
       ],
       "execution_count": null,
       "outputs": []
@@ -526,4 +526,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}
diff --git a/examples/41_Train_a_language_model_from_scratch.ipynb b/examples/41_Train_a_language_model_from_scratch.ipynb
index 625106cf..8bd4fc11 100644
--- a/examples/41_Train_a_language_model_from_scratch.ipynb
+++ b/examples/41_Train_a_language_model_from_scratch.ipynb
@@ -57,7 +57,7 @@
       "outputs": [],
       "source": [
         "%%capture\n",
-        "!pip install git+https://github.com/neuml/txtai datasets sentence-transformers onnxruntime onnx"
+        "!pip install git+https://github.com/neuml/txtai#egg=txtai[pipeline-train] datasets sentence-transformers onnxruntime onnx"
       ]
     },
     {
@@ -495,4 +495,4 @@
       }
     }
   ]
-}
\ No newline at end of file
+}