diff --git a/Project.toml b/Project.toml
index 6ea52754..f9aca992 100644
--- a/Project.toml
+++ b/Project.toml
@@ -58,7 +58,7 @@ SciMLBase = "2"
 SciMLSensitivity = "7.43"
 StableRNGs = "1.0.2"
 Statistics = "1.10"
-SteadyStateDiffEq = "2"
+SteadyStateDiffEq = "2.3.2"
 Test = "1.10"
 Zygote = "0.6.69"
 julia = "1.10"
diff --git a/test/layers_tests.jl b/test/layers_tests.jl
index 62d10592..3c810384 100644
--- a/test/layers_tests.jl
+++ b/test/layers_tests.jl
@@ -102,8 +102,8 @@ end
                        dense_layer(2 => 4) dense_layer(2 => 3) NoOpLayer() dense_layer(2 => 1);
                        dense_layer(1 => 4) dense_layer(1 => 3) dense_layer(1 => 2) NoOpLayer()]]
 
-    init_layers = [(dense_layer(4 => 4), dense_layer(4 => 3),
-        dense_layer(4 => 2), dense_layer(4 => 1))]
+    init_layers = [(
+        dense_layer(4 => 4), dense_layer(4 => 3), dense_layer(4 => 2), dense_layer(4 => 1))]
 
     x_sizes = [(4, 3)]
     scales = [((4,), (3,), (2,), (1,))]