diff --git a/src/blueair_api/device.py b/src/blueair_api/device.py
index 754a075..942968d 100644
--- a/src/blueair_api/device.py
+++ b/src/blueair_api/device.py
@@ -60,7 +60,10 @@ async def refresh(self):
         self.fan_speed = int(attributes["fan_speed"])
         self.filter_expired = attributes["filter_status"] != "OK"
         self.fan_mode = attributes["mode"]
-        self.wifi_working = attributes["wifi_status"] == "1"
+        if "wifi_status" in attributes:
+            self.wifi_working = attributes["wifi_status"] == "1"
+        else:
+            self.wifi_working = 0
         self.publish_updates()
 
     async def set_fan_speed(self, new_speed):