diff --git a/Project.toml b/Project.toml
index b2f6135..d5d4f08 100644
--- a/Project.toml
+++ b/Project.toml
@@ -28,6 +28,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 BoundaryValueDiffEq = "2"
 Convex = "0.15"
 DifferentialEquations = "7"
+Distributions = "0.25"
 Flux = "0.12, 0.13"
 Interpolations = "0.13"
 IntervalSets = "0.5, 0.7"