diff --git a/src/ReinforcementLearningFarm/Project.toml b/src/ReinforcementLearningFarm/Project.toml
index 2c20523e3..a5384a309 100644
--- a/src/ReinforcementLearningFarm/Project.toml
+++ b/src/ReinforcementLearningFarm/Project.toml
@@ -13,6 +13,7 @@ ReinforcementLearning = "158674fc-8238-5cab-b5ba-03dfc80d1318"
 
 [compat]
 FillArrays = "1"
+Flux = "0.14"
 CircularArrayBuffers = "0.1.12"
 Distributions = "0.25"
 ReinforcementLearning = "0.11"