diff --git a/src/inference/forward.jl b/src/inference/forward.jl
index 3e23c8a..653ce63 100644
--- a/src/inference/forward.jl
+++ b/src/inference/forward.jl
@@ -74,7 +74,7 @@ function forward!(
     seq_ends::AbstractVectorOrNTuple{Int},
     k::Integer,
 )
-    (; α, B, c) = storage
+    (; α, B, c, logL) = storage
     t1, t2 = seq_limits(seq_ends, k)
 
     # Initialization
diff --git a/src/inference/viterbi.jl b/src/inference/viterbi.jl
index 070beca..2eb495d 100644
--- a/src/inference/viterbi.jl
+++ b/src/inference/viterbi.jl
@@ -47,7 +47,7 @@ function viterbi!(
     seq_ends::AbstractVectorOrNTuple{Int},
     k::Integer,
 ) where {R}
-    (; q, logB, ϕ, ψ) = storage
+    (; q, logB, ϕ, ψ, logL) = storage
     t1, t2 = seq_limits(seq_ends, k)
 
     logBₜ₁ = view(logB, :, t1)