diff --git a/Project.toml b/Project.toml
index 878cef7d..32bf227c 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "InvertibleNetworks"
 uuid = "b7115f24-5f92-4794-81e8-23b0ddb121d3"
 authors = ["Philipp Witte <p.witte@ymail.com>", "Ali Siahkoohi <alisk@gatech.edu>", "Mathias Louboutin <mlouboutin3@gatech.edu>", "Gabrio Rizzuti <g.rizzuti@umcutrecht.nl>", "Rafael Orozco <rorozco@gatech.edu>", "Felix J. herrmann <fherrmann@gatech.edu>"]
-version = "2.2.3"
+version = "2.2.4"
 
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
diff --git a/src/utils/chainrules.jl b/src/utils/chainrules.jl
index 830ff75f..36456c71 100644
--- a/src/utils/chainrules.jl
+++ b/src/utils/chainrules.jl
@@ -70,7 +70,7 @@ function forward_update!(state::InvertibleOperationsTape, X::AbstractArray{T,N},
 
     if isa_newblock(state, X)
         push!(state.Y, Y)
-        push!(state.layer_blocks, [net])
+        push!(state.layer_blocks, Vector{Any}([net]))
         state.counter_block += 1
         state.counter_layer = 1
     else
diff --git a/src/utils/compute_utils.jl b/src/utils/compute_utils.jl
index c4b6dec8..330050fc 100644
--- a/src/utils/compute_utils.jl
+++ b/src/utils/compute_utils.jl
@@ -21,7 +21,7 @@ function chain_lr(x::AbstractMatrix{T}, vi::Vararg{AbstractVector{T}, N}) where
     out = T(1) .* x
     tmp = cuzeros(vi[1], size(x, 1))
     for v=vi
-        n = -2/norm(v)^2
+        n = -2/dot(v, v)
         mul!(tmp, out, v)
         rmul!(tmp, n)
         gemm_outer!(out, tmp, v)