diff --git a/glotaran/builtin/megacomplexes/decay/k_matrix.py b/glotaran/builtin/megacomplexes/decay/k_matrix.py
index 20280eaec..b47264ca2 100644
--- a/glotaran/builtin/megacomplexes/decay/k_matrix.py
+++ b/glotaran/builtin/megacomplexes/decay/k_matrix.py
@@ -121,9 +121,9 @@ def a_matrix_as_markdown(
             The initial concentration.
         """
         return self._array_as_markdown(
-            self.a_matrix(initial_concentration).T,
+            self.a_matrix(compartments, initial_concentration).T,
             compartments,
-            self.rates(initial_concentration),
+            self.rates(compartments, initial_concentration),
         )
 
     @staticmethod