diff --git a/custom_components/mcp_bridge/__init__.py b/custom_components/mcp_bridge/__init__.py
index 1000056..f16784c 100644
--- a/custom_components/mcp_bridge/__init__.py
+++ b/custom_components/mcp_bridge/__init__.py
@@ -30,7 +30,8 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
         exposed_entities = []
         
         # Iterate through all entities
-        for entity_id, state in hass.states.async_all():
+        for state in hass.states.async_all():
+            entity_id = state.entity_id
             entity_entry = entity_reg.async_get(entity_id)
             
             # Check if entity is exposed to conversation (voice assistant)