diff --git a/Project.toml b/Project.toml
index 07250d3..ba12464 100644
--- a/Project.toml
+++ b/Project.toml
@@ -28,7 +28,7 @@ NormalizingFlowsReverseDiffExt = "ReverseDiff"
 NormalizingFlowsZygoteExt = "Zygote"
 
 [compat]
-ADTypes = "0.1"
+ADTypes = "0.1, 0.2"
 Bijectors = "0.12.6, 0.13"
 DiffResults = "1"
 Distributions = "0.25"