diff --git a/research/src/riemannian_hmc.jl b/research/src/riemannian_hmc.jl
index 1bef6355..ab98185e 100644
--- a/research/src/riemannian_hmc.jl
+++ b/research/src/riemannian_hmc.jl
@@ -242,7 +242,7 @@ phasepoint(
     θ::T,
     r::T;
     ℓπ = ∂H∂θ(h, θ),
-    ℓκ = DualValue(neg_energy(h, r, θ), ∂H∂r(h, r, θ)),
+    ℓκ = DualValue(neg_energy(h, r, θ), ∂H∂r(h, θ, r)),
 ) where {T<:AbstractVecOrMat} = PhasePoint(θ, r, ℓπ, ℓκ)
 
 # Negative kinetic energy