diff --git a/ext/SymbolicsLuxCoreExt.jl b/ext/SymbolicsLuxCoreExt.jl
index 35613582a..e1d503060 100644
--- a/ext/SymbolicsLuxCoreExt.jl
+++ b/ext/SymbolicsLuxCoreExt.jl
@@ -4,7 +4,7 @@ using LuxCore, Symbolics
 
 @register_array_symbolic LuxCore.stateless_apply(
     model::LuxCore.AbstractLuxLayer, x::AbstractArray, ps::Union{NamedTuple, <:AbstractVector}) begin
-    size = LuxCore.outputsize(model, x, LuxCore._default_rng())
+    size = LuxCore.outputsize(model, x, LuxCore.Lux.Random())
     eltype = Real
 end