diff --git a/src/utilities.jl b/src/utilities.jl
index b31afdb4..5703c0da 100644
--- a/src/utilities.jl
+++ b/src/utilities.jl
@@ -1,5 +1,5 @@
 """
-    partition(rows::AbstractVector{Int}, fractions...; shuffle=false)
+    partition(rows::AbstractVector{Int}, fractions...; shuffle=false, rng=Random.GLOBAL_RNG)
 
 Splits the vector `rows` into a tuple of vectors whose lengths are
 given by the corresponding `fractions` of `length(rows)`. The last