diff --git a/Project.toml b/Project.toml
index 3c57e214..ae26decf 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "OptimalTransport"
 uuid = "7e02d93a-ae51-4f58-b602-d97af76e3b33"
 authors = ["zsteve <stephenz@student.unimelb.edu.au>"]
-version = "0.3.20"
+version = "0.3.21"
 
 [deps]
 ExactOptimalTransport = "24df6009-d856-477c-ac5c-91f668376b31"
@@ -15,7 +15,7 @@ Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 ExactOptimalTransport = "0.1, 0.2"
 IterativeSolvers = "0.8.4, 0.9"
 LogExpFunctions = "0.2, 0.3"
-NNlib = "0.6, 0.7, 0.8"
+NNlib = "0.6, 0.7, 0.8, 0.9"
 Reexport = "1"
 julia = "1"
 
diff --git a/docs/Project.toml b/docs/Project.toml
index f64b3c95..12af440f 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -5,5 +5,5 @@ OptimalTransport = "7e02d93a-ae51-4f58-b602-d97af76e3b33"
 
 [compat]
 ExactOptimalTransport = "0.1, 0.2"
-Documenter = "0.27"
+Documenter = "0.27, 1"
 OptimalTransport = "0.3"
diff --git a/examples/OneDimension/Project.toml b/examples/OneDimension/Project.toml
index 63fd8dbf..f9e446a0 100644
--- a/examples/OneDimension/Project.toml
+++ b/examples/OneDimension/Project.toml
@@ -12,4 +12,5 @@ Distances = "0.10"
 Distributions = "0.25"
 Literate = "2.9"
 OptimalTransport = "0.3"
+StatsPlots = "0.15"
 julia = "1"
diff --git a/examples/nmf/Project.toml b/examples/nmf/Project.toml
index 5058aab8..d27ed40a 100644
--- a/examples/nmf/Project.toml
+++ b/examples/nmf/Project.toml
@@ -13,14 +13,14 @@ Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 
 [compat]
-MLDatasets = "0.5"
-NNlib = "0.8"
-NMF = "0.5"
-LineSearches = "7"
-Optim = "1"
 Distances = "0.10"
+LineSearches = "7"
 Literate = "2"
-OptimalTransport = "0.3"
 LogExpFunctions = "0.3"
+MLDatasets = "0.5"
+NMF = "0.5, 1"
+NNlib = "0.8, 0.9"
+Optim = "1"
+OptimalTransport = "0.3"
 Plots = "1"
-StatsBase = "0.33"
+StatsBase = "0.33, 0.34"
diff --git a/examples/variational/Project.toml b/examples/variational/Project.toml
index 75d7d8f3..3fde61d0 100644
--- a/examples/variational/Project.toml
+++ b/examples/variational/Project.toml
@@ -18,5 +18,5 @@ Optim = "1.4"
 OptimalTransport = "0.3"
 Plots = "1"
 ReverseDiff = "1"
-StatsBase = "0.33"
+StatsBase = "0.33, 0.34"
 julia = "1"
diff --git a/test/gpu/Project.toml b/test/gpu/Project.toml
index 5b8296f4..d008a01f 100644
--- a/test/gpu/Project.toml
+++ b/test/gpu/Project.toml
@@ -3,6 +3,6 @@ CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"
 
 [compat]
-CUDA = "3"
+CUDA = "3, 4, 5"
 NNlibCUDA = "0.2"
 julia = "1.6"