diff --git a/docs/examples/Running-from-Spark-Shell.md b/docs/examples/Running-from-Spark-Shell.md
index 5132e22591..685cda4900 100644
--- a/docs/examples/Running-from-Spark-Shell.md
+++ b/docs/examples/Running-from-Spark-Shell.md
@@ -3,7 +3,7 @@
 Start up your spark shell and add the [TransmogrifAI package](https://spark-packages.org/package/salesforce/TransmogrifAI):
 
 ```bash
-$SPARK_HOME/bin/spark-shell --packages com.salesforce.transmogrifai:transmogrifai-core_2.11:0.3.4
+$SPARK_HOME/bin/spark-shell --packages com.salesforce.transmogrifai:transmogrifai-core_2.11:0.4.0
 ```
 
 Or if you'd like to use the latest version from master: