From 5fb344e5a2e32c0d4512a7750ef76ebae4f497e5 Mon Sep 17 00:00:00 2001
From: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Date: Thu, 2 Mar 2023 17:21:42 +0000
Subject: [PATCH] minor bugfix in tests

---
 test/utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/utils.jl b/test/utils.jl
index 2a1cc45..44ba890 100644
--- a/test/utils.jl
+++ b/test/utils.jl
@@ -22,7 +22,7 @@ function StateHistoryCallback(states, selector=deepcopy)
     return StateHistoryCallback{typeof(states), typeof(selector)}(states, selector)
 end
 
-function (cb::StateHistoryCallback)(rng, model, sampler, sample, state, i; kwargs...)
+function (cb::StateHistoryCallback)(rng, model, sampler, transition, state, iteration; kwargs...)
     push!(cb.states, cb.selector(state))
     return nothing
 end