From 8fc7d0b61e1b7cd3c9a24a331ef97830698f144b Mon Sep 17 00:00:00 2001
From: estevez-dev <vyalov.egor@gmail.com>
Date: Sun, 15 Sep 2019 14:34:00 +0300
Subject: [PATCH] Fix entity state non updated on entity page

---
 lib/pages/entity.page.dart | 3 ++-
 lib/pages/main.page.dart   | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/pages/entity.page.dart b/lib/pages/entity.page.dart
index 137238d..ca20189 100644
--- a/lib/pages/entity.page.dart
+++ b/lib/pages/entity.page.dart
@@ -21,7 +21,8 @@ class _EntityViewPageState extends State<EntityViewPage> {
     super.initState();
     _stateSubscription = eventBus.on<StateChangedEvent>().listen((event) {
       if (event.entityId == widget.entityId) {
-        Logger.d("State change event handled by entity page: ${event.entityId}");
+        entity = HomeAssistant().entities.get(widget.entityId);
+        Logger.d("[Entity page] State change event handled: ${event.entityId}");
         setState(() {});
       }
     });
diff --git a/lib/pages/main.page.dart b/lib/pages/main.page.dart
index f277325..b057162 100644
--- a/lib/pages/main.page.dart
+++ b/lib/pages/main.page.dart
@@ -661,7 +661,7 @@ class _MainPageState extends ReceiveShareState<MainPage> with WidgetsBindingObse
     Widget mediaMenuIcon;
     int playersCount = 0;
     if (!empty && !HomeAssistant().entities.isEmpty) {
-      List<Entity> activePlayers = HomeAssistant().entities.getByDomains(domains: ["media_player"], stateFiler: [EntityState.paused, EntityState.playing]);
+      List<Entity> activePlayers = HomeAssistant().entities.getByDomains(domains: ["media_player"], stateFiler: [EntityState.paused, EntityState.playing, EntityState.idle]);
       playersCount = activePlayers.length;
       mediaMenuItems.addAll(
           activePlayers.map((entity) => PopupMenuItem<String>(