diff --git a/src/MatrixSets/MatrixSets.jl b/src/MatrixSets/MatrixSets.jl
index 95f4b64102..39f93766d0 100644
--- a/src/MatrixSets/MatrixSets.jl
+++ b/src/MatrixSets/MatrixSets.jl
@@ -1,6 +1,7 @@
 module MatrixSets
 
 using ..LazySets
+import ..LazySets: linear_map
 export MatrixZonotope
 
 include("MatrixZonotope.jl")
diff --git a/src/MatrixSets/MatrixZonotope.jl b/src/MatrixSets/MatrixZonotope.jl
index baf588ea11..e7bf68cae1 100644
--- a/src/MatrixSets/MatrixZonotope.jl
+++ b/src/MatrixSets/MatrixZonotope.jl
@@ -1,8 +1,8 @@
-"""
+#="""
 See Def. 4 in [1].
 
 [1] Luo, Ertai, Niklas Kochdumper, and Stanley Bak. "Reachability analysis for linear systems with uncertain parameters using polynomial zonotopes." Proceedings of the 26th ACM International Conference on Hybrid Systems: Computation and Control. 2023.
-"""
+"""=#
 struct MatrixZonotope{N,MN<:AbstractMatrix{N}}
     "Center matrix."
     A0::MN
@@ -22,11 +22,11 @@ struct MatrixZonotope{N,MN<:AbstractMatrix{N}}
     end
 end
 
-"""
+#="""
 See Prop. 1 in [1].
 
 [1] Luo, Ertai, Niklas Kochdumper, and Stanley Bak. "Reachability analysis for linear systems with uncertain parameters using polynomial zonotopes." Proceedings of the 26th ACM International Conference on Hybrid Systems: Computation and Control. 2023.
-"""
+"""=#
 function linear_map(MZ::MatrixZonotope, P::SimpleSparsePolynomialZonotope)
     A0 = MZ.A0
     Ai = MZ.Ai