diff --git a/helper.jl b/helper.jl
index 0307729..ae1bb0e 100644
--- a/helper.jl
+++ b/helper.jl
@@ -25,5 +25,5 @@ nearest(list::Array, ref) = list[nearestIndex(list, ref)]
 "Simple implementation of the Kronecker sum"
 function kron_sum(A::AbstractMatrix, B::AbstractMatrix)
     @assert size(A, 1) == size(A, 2) && size(B, 1) == size(B, 2) "Matrices should be square"
-    return kron(A, I(size(B, 1))) + kron(B, I(size(A, 1)))
+    return kron(A, I(size(B, 1))) + kron(I(size(A, 1)), B)
 end
\ No newline at end of file