diff --git a/custom_components/xiaomi_miot/config_flow.py b/custom_components/xiaomi_miot/config_flow.py
index 9c448fc846..690fe84d85 100644
--- a/custom_components/xiaomi_miot/config_flow.py
+++ b/custom_components/xiaomi_miot/config_flow.py
@@ -522,7 +522,7 @@ async def async_step_customizing(self, user_input=None):
                 for v in self.hass.data[DOMAIN].values():
                     if isinstance(v, dict):
                         if mod := v.get('miio_info', {}).get(CONF_MODEL):
-                            models.append(mod)
+                            models[mod] = v
                         v = v.get(CONF_XIAOMI_CLOUD)
                     if isinstance(v, MiotCloud):
                         mic = v