diff --git a/homeassistant/components/sensor/steam_online.py b/homeassistant/components/sensor/steam_online.py index 1264e9a8556..6b740d0639b 100644 --- a/homeassistant/components/sensor/steam_online.py +++ b/homeassistant/components/sensor/steam_online.py @@ -46,26 +46,20 @@ class SteamSensor(Entity): @property def state(self): """ State of the sensor. """ - if self._profile.status == 1: - account_state = 'Online' - elif self._profile.status == 2: - account_state = 'Busy' - elif self._profile.status == 3: - account_state = 'Away' - elif self._profile.status == 4: - account_state = 'Snooze' - elif self._profile.status == 5: - account_state = 'Trade' - elif self._profile.status == 6: - account_state = 'Play' - else: - account_state = 'Offline' - return account_state + return self._state # pylint: disable=no-member def update(self): """ Update device state. """ self._profile = self._steamod.user.profile(self._account) + self._state = { + 1: 'Online', + 2: 'Busy', + 3: 'Away', + 4: 'Snooze', + 5: 'Trade', + 6: 'Play', + }.get(self._profile.status, 'Offline') @property def device_state_attributes(self):