diff --git a/ext/AdvancedVIEnzymeExt.jl b/ext/AdvancedVIEnzymeExt.jl
index 55f22897..5bd2aa0a 100644
--- a/ext/AdvancedVIEnzymeExt.jl
+++ b/ext/AdvancedVIEnzymeExt.jl
@@ -17,7 +17,7 @@ function AdvancedVI.value_and_gradient!(
     ∇θ = DiffResults.gradient(out)
     fill!(∇θ, zero(T))
     _, y = Enzyme.autodiff(
-        Enzyme.ReverseWithPrimal, f, Enzyme.Active, Enzyme.Duplicated(θ, ∇θ)
+        Enzyme.ReverseWithPrimal, Enzyme.Const(f), Enzyme.Active, Enzyme.Duplicated(θ, ∇θ)
     )
     DiffResults.value!(out, y)
     return out