diff --git a/pyaml/diagnostics/chromaticity_monitor.py b/pyaml/diagnostics/chromaticity_monitor.py index ab60b6a6..550749ed 100644 --- a/pyaml/diagnostics/chromaticity_monitor.py +++ b/pyaml/diagnostics/chromaticity_monitor.py @@ -224,14 +224,14 @@ def measure( # Get devices self.check_peer() - tm = self._peer.get_betatron_tune_monitor(self._cfg.betatron_tune_name) - rf = self._peer.get_rf_plant(self._cfg.rf_plant_name) + tm = self.peer.get_betatron_tune_monitor(self._cfg.betatron_tune_name) + rf = self.peer.get_rf_plant(self._cfg.rf_plant_name) bpms = None n_bpm = 0 orbit = None if fit_dispersion and fit_disp_order is not None and self._cfg.bpm_array_name is not None: # For dispersion fit - bpms = self._peer.get_bpms(self._cfg.bpm_array_name) + bpms = self.peer.get_bpms(self._cfg.bpm_array_name) n_bpm = len(bpms) f0 = rf.frequency.get() diff --git a/pyaml/tuning_tools/chromaticity.py b/pyaml/tuning_tools/chromaticity.py index c1d3dea0..a06ab569 100644 --- a/pyaml/tuning_tools/chromaticity.py +++ b/pyaml/tuning_tools/chromaticity.py @@ -95,12 +95,12 @@ def load(self, load_path: Path): @property def _cm(self) -> "ChomaticityMonitor": self.check_peer() - return self._peer.get_chromaticity_monitor(self._cfg.chromaticty_monitor_name) + return self.peer.get_chromaticity_monitor(self._cfg.chromaticty_monitor_name) @property def _sextu(self) -> "MagnetArray": self.check_peer() - return self._peer.get_magnets(self._cfg.sextu_array_name) + return self.peer.get_magnets(self._cfg.sextu_array_name) def get(self): """