diff --git a/Project.toml b/Project.toml
index ddd51648..a0cdb901 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "DiffEqBayes"
 uuid = "ebbdde9d-f333-5424-9be2-dbf1e9acfb5e"
 authors = ["Vaibhavdixit02 <vaibhavyashdixit@gmail.com>"]
-version = "3.0.0"
+version = "3.0.1"
 
 [deps]
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
diff --git a/src/dynamichmc_inference.jl b/src/dynamichmc_inference.jl
index 17a41ec2..15d7833a 100644
--- a/src/dynamichmc_inference.jl
+++ b/src/dynamichmc_inference.jl
@@ -114,5 +114,11 @@ function dynamichmc_inference(problem::DiffEqBase.DEProblem, algorithm, t, data,
     ℓ = TransformedLogDensity(trans, P)
     ∇ℓ = LogDensityProblems.ADgradient(AD_gradient_kind, ℓ)
     results = mcmc_with_warmup(rng, ∇ℓ, num_samples; mcmc_kwargs...)
-    merge((posterior = TransformVariables.transform.(Ref(trans), results.chain),), results)
+    chain = if haskey(results, :chain) # DynamicHMC < 3.3.0
+        results.chain
+    else
+        eachcol(results.posterior_matrix)
+    end
+    posterior = map(Base.Fix1(TransformVariables.transform, trans), chain)
+    merge((; posterior), results)
 end