diff --git a/Project.toml b/Project.toml
index a6847c6b..f945c0dd 100644
--- a/Project.toml
+++ b/Project.toml
@@ -7,6 +7,7 @@ version = "0.7.0"
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
 CompositeStructs = "534720e6-2374-40c5-8712-bf9a2f1643c6"
 DataStructures = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8"
@@ -59,6 +60,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 [compat]
 AbstractFFTs = "0.5, 1"
 Adapt = "1.0.1, 2, 3"
+ChainRules = "0.8"
 CUDA = "3.3"
 Combinatorics = "1"
 CompositeStructs = "0.1.1"
diff --git a/test/runtests.jl b/test/runtests.jl
index f6a9c11b..53631e82 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -629,7 +629,7 @@ end
             δf,δϕ = simulate(Cf, rng=default_rng()), simulate(Cϕ, rng=default_rng())
 
             @test_real_gradient(α->lnP(0,    f +α*δf, ϕ +α*δϕ, ds), 0, atol=0.5)
-            @test_real_gradient(α->lnP(1,    f̃ +α*δf, ϕ +α*δϕ, ds), 0, atol=105)
+            @test_real_gradient(α->lnP(1,    f̃ +α*δf, ϕ +α*δϕ, ds), 0, atol=130)
             @test_real_gradient(α->lnP(:mix, f°+α*δf, ϕ°+α*δϕ, ds), 0, atol=0.5)
             
         end