From bb0b5ed3ad0d03da43af708176557a5f97b97f25 Mon Sep 17 00:00:00 2001
From: Gamelin Alexis <alexis.gamelin@synchrotron-soleil.fr>
Date: Mon, 20 Feb 2023 17:17:24 +0100
Subject: [PATCH] [Fix] ExponentialDamper

---
 mbtrack2/tracking/feedback.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/mbtrack2/tracking/feedback.py b/mbtrack2/tracking/feedback.py
index c16afbd..5e4e5f2 100644
--- a/mbtrack2/tracking/feedback.py
+++ b/mbtrack2/tracking/feedback.py
@@ -30,15 +30,12 @@ class ExponentialDamper(Element):
         self.plane = plane
         if self.plane == "x":
             self.action = "xp"
-            self.damp_idx = 0
             self.mean_idx = 1
         elif self.plane == "y":
             self.action = "yp"
-            self.damp_idx = 1
             self.mean_idx = 3
         elif self.plane == "s":
             self.action = "delta"
-            self.damp_idx = 2
             self.mean_idx = 5
         else:
             raise ValueError(f"plane should be x, y or s, not {self.plane}")
@@ -55,7 +52,7 @@ class ExponentialDamper(Element):
         bunch : Bunch or Beam object
         """
         bunch[self.action] -= (2*self.ring.T0/
-                               self.damping_time[self.damp_idx]*
+                               self.damping_time*
                                np.sin(self.phase_diff)*
                                bunch.mean[self.mean_idx])
 
-- 
GitLab