diff --git a/mbtrack2/tracking/particles.py b/mbtrack2/tracking/particles.py index 8a50004481ffe1003c7ca7014cd95b69f6a9c4a4..db333b346e7410db5204a61f81980ba37bf9c659 100644 --- a/mbtrack2/tracking/particles.py +++ b/mbtrack2/tracking/particles.py @@ -288,7 +288,7 @@ class Bunch: cov_y = np.cov(self['y'], self['yp']) cov_z = np.cov(self['tau'], self['delta']) - if (self.ring.optics.local_dispersion != 0).any(): + if (self.ring.optics.local_dispersion != [0, 0, 0, 0]): cov_xdelta = np.cov(self['x'], self['delta']) cov_xpdelta = np.cov(self['xp'], self['delta']) cov_ydelta = np.cov(self['y'], self['delta'])