Cleanup group areas

1 files changed, 2 insertions, 6 deletions

diff --git a/alarmpy/alarmpy.py b/alarmpy/alarmpy.py
index 90122b7..0edec3a 100644
--- a/alarmpy/alarmpy.py
+++ b/alarmpy/alarmpy.py
@@ -134,12 +134,8 @@ class Alarm:
     def group_areas_and_localize(self, cities):
         res = defaultdict(list)
         for city in cities:
-            try:
-                area = self.labels[city][f"areaname_{self.language}"]
-                label = self.labels[city][f"label_{self.language}"]
-            except KeyError:
-                area = ""
-                label = city
+            area = self.labels.get(city, {}).get(f"areaname_{self.language}", "")
+            label = self.labels.get(city, {}).get(f"label_{self.language}", city)
             res[area].append(label)
         return res