diff --git a/src/sampler.jl b/src/sampler.jl
index 67925be3e..5c583acd8 100644
--- a/src/sampler.jl
+++ b/src/sampler.jl
@@ -87,9 +87,9 @@ function AbstractMCMC.sample(
     N::Integer;
     chain_type=default_chain_type(sampler),
     resume_from=nothing,
+    initial_state = loadstate(resume_from),
     kwargs...,
 )
-    initial_state = loadstate(resume_from)
     return AbstractMCMC.mcmcsample(
         rng, model, sampler, N; chain_type, initial_state, kwargs...
     )