From 8d77d7812db208032e334c2e989e0158624f1fa7 Mon Sep 17 00:00:00 2001
From: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Date: Fri, 3 Feb 2023 12:15:35 +0000
Subject: [PATCH] fixed test_utils

---
 src/test_utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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