From cf63f75e50ecc8ac844610e7c671be8e03e8ad71 Mon Sep 17 00:00:00 2001
From: Fredrik Bagge Carlson <baggepinnen@gmail.com>
Date: Fri, 14 Jan 2022 11:13:34 +0100
Subject: [PATCH] fix returned transformation matrix

---
 src/matrix_comps.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/matrix_comps.jl b/src/matrix_comps.jl
index 748aee745..c20d125d1 100644
--- a/src/matrix_comps.jl
+++ b/src/matrix_comps.jl
@@ -513,7 +513,8 @@ function balreal(sys::ST) where ST <: AbstractStateSpace
     L = lmul!(R',view(U,:,i1))*hsi2
     Tr = lmul!(S,V[:,i1])*hsi2
     # return the minimal balanced system
-    return ss(L'SF.T*Tr, L'bs, cs*Tr, sys.D, sys.timeevol), Diagonal(Σ), L
+    T = L'SF.Z'
+    return ss(L'SF.T*Tr, L'bs, cs*Tr, sys.D, sys.timeevol), Diagonal(Σ), T
 end