diff --git a/tracking/optics.py b/tracking/optics.py index da6ae20aec203e9828cf92d9066fda9f79345357..0f16258355fe4abed9d4e1b08c50eec39189215c 100644 --- a/tracking/optics.py +++ b/tracking/optics.py @@ -66,11 +66,11 @@ class Optics: self.use_local_values = False self.load_from_AT(lattice_file, **kwargs) if local_beta is None: - self.local_beta = np.mean(self.beta_array) + self.local_beta = np.mean(self.beta_array, axis=1) else: self.local_beta = local_beta if local_alpha is None: - self.local_alpha = np.mean(self.alpha_array) + self.local_alpha = np.mean(self.alpha_array, axis=1) else: self.local_alpha = local_alpha if local_dispersion is None: