diff --git a/src/ReinforcementLearningCore/Project.toml b/src/ReinforcementLearningCore/Project.toml
index 4711550ae..1ac4e98b9 100644
--- a/src/ReinforcementLearningCore/Project.toml
+++ b/src/ReinforcementLearningCore/Project.toml
@@ -32,7 +32,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 [compat]
 AbstractTrees = "0.3"
 Adapt = "2, 3"
-ArrayInterface = "3"
+ArrayInterface = "3, 4"
 CUDA = "3.5"
 CircularArrayBuffers = "0.1"
 Compat = "3"