diff --git a/tests/components/mobile_app/test_entity.py b/tests/components/mobile_app/test_entity.py
index d8cb91a8bc6..f399f842745 100644
--- a/tests/components/mobile_app/test_entity.py
+++ b/tests/components/mobile_app/test_entity.py
@@ -37,11 +37,8 @@ async def test_sensor(hass, create_registrations, webhook_client):  # noqa: F401
     json = await reg_resp.json()
     assert json == {'status': 'registered'}
 
-    # 3 because we require device_tracker which adds zone.home and
-    # group.all_devices
-    assert len(hass.states.async_all()) == 3
-
-    entity = hass.states.async_all()[2]
+    entity = hass.states.get('sensor.battery_state')
+    assert entity is not None
 
     assert entity.attributes['device_class'] == 'battery'
     assert entity.attributes['icon'] == 'mdi:battery'