diff --git a/Project.toml b/Project.toml
index 4c40a85dd..d0e5f755f 100644
--- a/Project.toml
+++ b/Project.toml
@@ -40,7 +40,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
 Adapt = "4"
-AdvancedHMC = "0.6"
+AdvancedHMC = "0.6.1"
 Aqua = "0.8"
 ArrayInterface = "7.7"
 CUDA = "5.2"