fix error found with test

firstof9 · Jul 6, 2024 · 7901141 · 7901141
1 parent f7b9fa6
commit 7901141
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/mail_and_packages/config_flow.py b/custom_components/mail_and_packages/config_flow.py
@@ -457,8 +457,12 @@ async def async_step_reconfig_2(self, user_input=None):
             self._errors, user_input = await _validate_user_input(user_input)
             self._data.update(user_input)
             if len(self._errors) == 0:
+                if any(
+                    sensor in self._data[CONF_RESOURCES] for sensor in AMAZON_SENSORS
+                ):
+                    return await self.async_step_config_amazon()
                 if self._data[CONF_CUSTOM_IMG]:
-                    return await self.async_step_reconfig_3()
+                    return await self.async_step_config_3()
                 self.hass.config_entries.async_update_entry(
                     self._entry, data=self._data
                 )