fixing problem

mampfes · Nov 28, 2024 · b9971cc · b9971cc
1 parent 4b193a0
commit b9971cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/custom_components/epex_spot/config_flow.py b/custom_components/epex_spot/config_flow.py
@@ -27,7 +27,7 @@
     DEFAULT_TAX,
     DOMAIN,
 )
-from .EPEXSpot import SMARD, Awattar, EPEXSpotWeb, Tibber, smartENERGY
+from .EPEXSpot import SMARD, Awattar, EPEXSpotWeb, Tibber, smartENERGY, Energyforecast
 
 CONF_SOURCE_LIST = (
     CONF_SOURCE_AWATTAR,
@@ -83,6 +83,7 @@ async def async_step_source(self, user_input=None):
             )
         # Energyforecast API requires a token
         elif self._source_name == CONF_SOURCE_ENERGYFORECAST:
+            areas = Energyforecast.Energyforecast.MARKET_AREAS
             data_schema = vol.Schema(
                 {
                     vol.Required(CONF_MARKET_AREA): vol.In(sorted(areas)),