diff --git a/Project.toml b/Project.toml
index 6714b1ddb..0b8dc927b 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.22.0"
+version = "0.22.1"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"
@@ -24,7 +24,7 @@ ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
 AbstractMCMC = "2, 3.0, 4"
 AbstractPPL = "0.5.3"
 BangBang = "0.3"
-Bijectors = "0.11"
+Bijectors = "0.11, 0.12"
 ChainRulesCore = "0.9.7, 0.10, 1"
 ConstructionBase = "1"
 Distributions = "0.23.8, 0.24, 0.25"
diff --git a/test/Project.toml b/test/Project.toml
index 29bf12ebb..08556aa86 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -23,7 +23,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 [compat]
 AbstractMCMC = "2.1, 3.0, 4"
 AbstractPPL = "0.5.1, 0.6"
-Bijectors = "0.11"
+Bijectors = "0.11, 0.12"
 Distributions = "0.25"
 DistributionsAD = "0.6.3"
 Documenter = "0.26.1, 0.27"
@@ -33,6 +33,6 @@ MCMCChains = "4.0.4, 5"
 MacroTools = "0.5.5"
 Setfield = "0.7.1, 0.8, 1"
 StableRNGs = "1"
-Tracker = "0.2.11"
+Tracker = "0.2.23"
 Zygote = "0.5.4, 0.6"
 julia = "1.6"
diff --git a/test/test_util.jl b/test/test_util.jl
index 667412bca..f3e54c437 100644
--- a/test/test_util.jl
+++ b/test/test_util.jl
@@ -13,11 +13,8 @@ function test_model_ad(model, logp_manual)
     x = DynamicPPL.getall(vi)
 
     # Log probabilities using the model.
-    function logp_model(x)
-        new_vi = VarInfo(vi, SampleFromPrior(), x)
-        model(new_vi)
-        return getlogp(new_vi)
-    end
+    ℓ = DynamicPPL.LogDensityFunction(model, vi)
+    logp_model = Base.Fix1(LogDensityProblems.logdensity, ℓ)
 
     # Check that both functions return the same values.
     lp = logp_manual(x)