diff --git a/Project.toml b/Project.toml
index 008a5fdd1..6caa09922 100644
--- a/Project.toml
+++ b/Project.toml
@@ -40,7 +40,7 @@ NNlib = "0.8"
 NNlibCUDA = "0.2"
 NearestNeighbors = "0.4"
 Reexport = "1"
-StatsBase = "0.33"
+StatsBase = "0.33, 0.34"
 julia = "1.7"
 
 [extras]