From 955307d6974975d44cab5839bb87691ed1b71af2 Mon Sep 17 00:00:00 2001
From: Watanyu Foosang <watanyu.f@gmail.com>
Date: Tue, 9 Mar 2021 16:28:22 +0100
Subject: [PATCH] Minor fix SynchrotronRadiation

---
 tracking/element.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tracking/element.py b/tracking/element.py
index 6f23f7d..34fdc76 100644
--- a/tracking/element.py
+++ b/tracking/element.py
@@ -135,7 +135,7 @@ class SynchrotronRadiation(Element):
         if (self.switch[2] == True):
             rand = np.random.normal(size=len(bunch))
             bunch["yp"] = ((1 - 2*self.ring.T0/self.ring.tau[1])*bunch["yp"] +
-                 2*self.ring.sigma()[1]*(self.ring.T0/self.ring.tau[1])**0.5*rand)
+                 2*self.ring.sigma()[3]*(self.ring.T0/self.ring.tau[1])**0.5*rand)
         
 class TransverseMap(Element):
     """
-- 
GitLab