diff --git a/lib/pages/entity.page.dart b/lib/pages/entity.page.dart
index 6370f4a..2ac545b 100644
--- a/lib/pages/entity.page.dart
+++ b/lib/pages/entity.page.dart
@@ -35,7 +35,7 @@ class _EntityViewPageState extends State<EntityViewPage> {
   }
 
   _getEntity() {
-    _entity = HomeAssistant().entities.get(widget.entityId);
+    _entity = HomeAssistant().entities?.get(widget.entityId);
   }
 
   @override