diff --git a/homeassistant/components/wled/light.py b/homeassistant/components/wled/light.py
index 65866935562..4326f1066c7 100644
--- a/homeassistant/components/wled/light.py
+++ b/homeassistant/components/wled/light.py
@@ -86,9 +86,8 @@ async def async_setup_entry(
 
     update_segments = partial(
         async_update_segments,
-        entry,
         coordinator,
-        {},
+        set(),
         async_add_entities,
     )
 
@@ -363,30 +362,24 @@ class WLEDSegmentLight(WLEDEntity, LightEntity):
 
 @callback
 def async_update_segments(
-    entry: ConfigEntry,
     coordinator: WLEDDataUpdateCoordinator,
-    current: dict[int, WLEDSegmentLight | WLEDMasterLight],
+    current_ids: set[int],
     async_add_entities,
 ) -> None:
     """Update segments."""
     segment_ids = {light.segment_id for light in coordinator.data.state.segments}
-    current_ids = set(current)
-    new_entities = []
-
-    # Discard master (if present)
-    current_ids.discard(-1)
-
-    # Process new segments, add them to Home Assistant
-    for segment_id in segment_ids - current_ids:
-        current[segment_id] = WLEDSegmentLight(coordinator, segment_id)
-        new_entities.append(current[segment_id])
+    new_entities: list[WLEDMasterLight | WLEDSegmentLight] = []
 
     # More than 1 segment now? No master? Add master controls
     if not coordinator.keep_master_light and (
         len(current_ids) < 2 and len(segment_ids) > 1
     ):
-        current[-1] = WLEDMasterLight(coordinator)
-        new_entities.append(current[-1])
+        new_entities.append(WLEDMasterLight(coordinator))
+
+    # Process new segments, add them to Home Assistant
+    for segment_id in segment_ids - current_ids:
+        current_ids.add(segment_id)
+        new_entities.append(WLEDSegmentLight(coordinator, segment_id))
 
     if new_entities:
         async_add_entities(new_entities)
diff --git a/homeassistant/components/wled/select.py b/homeassistant/components/wled/select.py
index 845ff38b5e6..373565b7ef7 100644
--- a/homeassistant/components/wled/select.py
+++ b/homeassistant/components/wled/select.py
@@ -31,7 +31,7 @@ async def async_setup_entry(
     update_segments = partial(
         async_update_segments,
         coordinator,
-        {},
+        set(),
         async_add_entities,
     )
     coordinator.async_add_listener(update_segments)
@@ -118,19 +118,18 @@ class WLEDPaletteSelect(WLEDEntity, SelectEntity):
 @callback
 def async_update_segments(
     coordinator: WLEDDataUpdateCoordinator,
-    current: dict[int, WLEDPaletteSelect],
+    current_ids: set[int],
     async_add_entities,
 ) -> None:
     """Update segments."""
     segment_ids = {segment.segment_id for segment in coordinator.data.state.segments}
-    current_ids = set(current)
 
     new_entities = []
 
     # Process new segments, add them to Home Assistant
     for segment_id in segment_ids - current_ids:
-        current[segment_id] = WLEDPaletteSelect(coordinator, segment_id)
-        new_entities.append(current[segment_id])
+        current_ids.add(segment_id)
+        new_entities.append(WLEDPaletteSelect(coordinator, segment_id))
 
     if new_entities:
         async_add_entities(new_entities)