From 89d5b86c2cfab4d5b0583a915468463065e1e67d Mon Sep 17 00:00:00 2001
From: Joshua Lampert <joshua.lampert@uni-hamburg.de>
Date: Wed, 11 Sep 2024 13:26:27 +0200
Subject: [PATCH] use functions from analysis_integrals

---
 src/visualization.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/visualization.jl b/src/visualization.jl
index 7918918c..f4c44d39 100644
--- a/src/visualization.jl
+++ b/src/visualization.jl
@@ -186,10 +186,10 @@ end
 
         for (i, (name, integral)) in enumerate(pairs(ints))
             name in exclude && continue
-            name_function = getfield(@__MODULE__, name)
+            quantity = cb.affect!.analysis_integrals[i]
             @series begin
                 subplot --> subplot
-                label := pretty_form_utf(name_function) * " " * label_extension
+                label := pretty_form_utf(quantity) * " " * label_extension
                 title --> "change of invariants"
                 xguide --> "t"
                 yguide --> "change of invariants"