Merge branch 'marq24:feature/eController' into feature/eController

marq24 · Oct 20, 2024 · efee0f3 · efee0f3
2 parents c0ab230 + b0d12b7
commit efee0f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/goecharger_api2/config_flow.py b/custom_components/goecharger_api2/config_flow.py
@@ -221,7 +221,7 @@ async def async_step_user_lan(self, user_input=None):
             return await self._update_options()
 
         # is this the 11kW or the 22kW Version?
-        if int(self.options.get(CONF_MODEL)) == 11:
+        if self.options.get(CONF_INTEGRATION_TYPE, INTG_TYPE.CHARGER.value) == INTG_TYPE.CONTROLLER.value or int(self.options.get(CONF_MODEL)) == 11:
             return self.async_show_form(
                 step_id=step_type,
                 data_schema=vol.Schema({
@@ -247,7 +247,7 @@ async def async_step_user_wan(self, user_input=None):
             return await self._update_options()
 
         # is this the 11kW or the 22kW Version?
-        if int(self.options.get(CONF_MODEL)) == 11:
+        if self.options.get(CONF_INTEGRATION_TYPE, INTG_TYPE.CHARGER.value) == INTG_TYPE.CONTROLLER.value or int(self.options.get(CONF_MODEL)) == 11:
             return self.async_show_form(
                 step_id=step_type,
                 data_schema=vol.Schema({