diff --git a/benchmark/Project.toml b/benchmark/Project.toml
index b218270a..eb9e3336 100644
--- a/benchmark/Project.toml
+++ b/benchmark/Project.toml
@@ -12,7 +12,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 ADTypes = "0.2"
 BenchmarkTools = "1"
 ComponentArrays = "0.15"
-DifferentiationInterface = "0.1"
+DifferentiationInterface = "0.1, 0.2"
 Lux = "0.5"
 PkgBenchmark = "0.2"
 StableRNGs = "1"