From e581d9e249f2d543dece08176ba1fb194fbb5fa4 Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Fri, 15 Mar 2019 09:14:20 -0700 Subject: [PATCH] Fix more test --- tests/components/mobile_app/test_entity.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/components/mobile_app/test_entity.py b/tests/components/mobile_app/test_entity.py index f399f842745..5dc285cfe9e 100644 --- a/tests/components/mobile_app/test_entity.py +++ b/tests/components/mobile_app/test_entity.py @@ -65,8 +65,7 @@ async def test_sensor(hass, create_registrations, webhook_client): # noqa: F401 assert update_resp.status == 200 - updated_entity = hass.states.async_all()[2] - + updated_entity = hass.states.get('sensor.battery_state') assert updated_entity.state == '123'