diff --git a/Project.toml b/Project.toml
index 6a6a51ab..bc7a3cd7 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "MCMCDiagnosticTools"
 uuid = "be115224-59cd-429b-ad48-344e309966f0"
 authors = ["David Widmann"]
-version = "0.3.5"
+version = "0.3.6"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"
diff --git a/docs/Project.toml b/docs/Project.toml
index adc9a2d9..36d0a73f 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -4,11 +4,13 @@ EvoTrees = "f6006082-12f8-11e9-0c9c-0d5d367ab1e5"
 MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
 MLJIteration = "614be32b-d00c-4edb-bd02-1eb411ab5e55"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+StatisticalMeasures = "a19d573c-0a75-4610-95b3-7071388c7541"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [compat]
 Documenter = "0.27"
 EvoTrees = "0.14.7, 0.15, 0.16"
-MLJBase = "0.19, 0.20, 0.21"
-MLJIteration = "0.5"
+MLJBase = "0.19, 0.20, 0.21, 1"
+MLJIteration = "0.5, 0.6"
+StatisticalMeasures = "0.1"
 julia = "1.6"
diff --git a/src/rstar.jl b/src/rstar.jl
index 9b16b299..8a2fb26d 100644
--- a/src/rstar.jl
+++ b/src/rstar.jl
@@ -156,7 +156,7 @@ is returned (algorithm 2).
 # Examples
 
 ```jldoctest rstar; setup = :(using Random; Random.seed!(101))
-julia> using MLJBase, MLJIteration, EvoTrees, Statistics
+julia> using MLJBase, MLJIteration, EvoTrees, Statistics, StatisticalMeasures
 
 julia> samples = fill(4.0, 100, 3, 2);
 ```
diff --git a/test/Project.toml b/test/Project.toml
index 5751c756..d1674c81 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -27,9 +27,9 @@ EvoTrees = "0.14.7, 0.15"
 FFTW = "1.1"
 LogDensityProblems = "0.12, 1, 2"
 LogExpFunctions = "0.3"
-MLJBase = "0.19, 0.20, 0.21"
+MLJBase = "0.19, 0.20, 0.21, 1"
 MLJDecisionTreeInterface = "0.3, 0.4"
-MLJIteration = "0.5"
+MLJIteration = "0.5, 0.6"
 MLJLIBSVMInterface = "0.2"
 MLJModels = "0.16"
 MLJXGBoostInterface = "0.3"