diff --git a/src/sample.jl b/src/sample.jl
index df76caf0..9d48df39 100644
--- a/src/sample.jl
+++ b/src/sample.jl
@@ -44,9 +44,10 @@ convergence criterion `isdone` returns `true`, and return the samples.
 
 The function `isdone` has the signature
 ```julia
-isdone(rng, model, sampler, samples, iteration; kwargs...)
+isdone(rng, model, sampler, samples, state, iteration; kwargs...)
 ```
-and should return `true` when sampling should end, and `false` otherwise.
+where `state` and `iteration` are the current state and iteration of the sampler, respectively.
+It should return `true` when sampling should end, and `false` otherwise.
 """
 function StatsBase.sample(
     rng::Random.AbstractRNG,