diff --git a/homeassistant/components/vacuum/neato.py b/homeassistant/components/vacuum/neato.py index 53f83a1de1d..83a4ad7c58d 100644 --- a/homeassistant/components/vacuum/neato.py +++ b/homeassistant/components/vacuum/neato.py @@ -108,27 +108,28 @@ class NeatoConnectedVacuum(StateVacuumDevice): self._clean_state = STATE_ERROR self._status_state = ERRORS.get(self._state['error']) - if not self._mapdata.get(self.robot.serial, {}).get('maps', []): + if not self._mapdata.get(self._robot_serial, {}).get('maps', []): return self.clean_time_start = ( - (self._mapdata[self.robot.serial]['maps'][0]['start_at'] + (self._mapdata[self._robot_serial]['maps'][0]['start_at'] .strip('Z')) .replace('T', ' ')) self.clean_time_stop = ( - (self._mapdata[self.robot.serial]['maps'][0]['end_at'].strip('Z')) + (self._mapdata[self._robot_serial]['maps'][0]['end_at'].strip('Z')) .replace('T', ' ')) self.clean_area = ( - self._mapdata[self.robot.serial]['maps'][0]['cleaned_area']) + self._mapdata[self._robot_serial]['maps'][0]['cleaned_area']) self.clean_suspension_charge_count = ( - self._mapdata[self.robot.serial]['maps'][0] + self._mapdata[self._robot_serial]['maps'][0] ['suspended_cleaning_charging_count']) self.clean_suspension_time = ( - self._mapdata[self.robot.serial]['maps'][0] + self._mapdata[self._robot_serial]['maps'][0] ['time_in_suspended_cleaning']) self.clean_battery_start = ( - self._mapdata[self.robot.serial]['maps'][0]['run_charge_at_start']) + self._mapdata[self._robot_serial]['maps'][0]['run_charge_at_start'] + ) self.clean_battery_end = ( - self._mapdata[self.robot.serial]['maps'][0]['run_charge_at_end']) + self._mapdata[self._robot_serial]['maps'][0]['run_charge_at_end']) self._battery_level = self._state['details']['charge']