diff --git a/src/test_utils.jl b/src/test_utils.jl
index 5dc5b1685..45b9fff07 100644
--- a/src/test_utils.jl
+++ b/src/test_utils.jl
@@ -686,7 +686,7 @@ Simple model for which [`default_transformation`](@ref) returns a [`StaticTransf
 end
 
 function DynamicPPL.default_transformation(::Model{typeof(demo_static_transformation)})
-    b = Bijectors.stack(exp, identity)
+    b = Bijectors.stack(Bijectors.elementwise(exp), identity)
     return DynamicPPL.StaticTransformation(b)
 end