diff --git a/test/gpu/Project.toml b/test/gpu/Project.toml
index fc7bf33c5..1763c5871 100644
--- a/test/gpu/Project.toml
+++ b/test/gpu/Project.toml
@@ -7,5 +7,5 @@ LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 [compat]
 CUDA = "3.12, 4, 5"
 DiffEqCallbacks = "2.24, 3"
-DiffEqFlux = "3"
+DiffEqFlux = "3, 4"
 LuxCUDA = "0.3.1"