diff --git a/homeassistant/components/amberelectric/config_flow.py b/homeassistant/components/amberelectric/config_flow.py
index efb5ddfb931..0258fdf4cb4 100644
--- a/homeassistant/components/amberelectric/config_flow.py
+++ b/homeassistant/components/amberelectric/config_flow.py
@@ -1,8 +1,6 @@
 """Config flow for the Amber Electric integration."""
 from __future__ import annotations
 
-from typing import Any
-
 import amberelectric
 from amberelectric.api import amber_api
 from amberelectric.model.site import Site
@@ -10,6 +8,7 @@ import voluptuous as vol
 
 from homeassistant import config_entries
 from homeassistant.const import CONF_API_TOKEN
+from homeassistant.data_entry_flow import FlowResult
 
 from .const import CONF_SITE_ID, CONF_SITE_NAME, CONF_SITE_NMI, DOMAIN
 
@@ -44,7 +43,9 @@ class AmberElectricConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
                 self._errors[CONF_API_TOKEN] = "unknown_error"
             return None
 
-    async def async_step_user(self, user_input: dict[str, Any] | None = None):
+    async def async_step_user(
+        self, user_input: dict[str, str] | None = None
+    ) -> FlowResult:
         """Step when user initializes a integration."""
         self._errors = {}
         self._sites = None
@@ -76,11 +77,14 @@ class AmberElectricConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
             errors=self._errors,
         )
 
-    async def async_step_site(self, user_input: dict[str, Any] = None):
+    async def async_step_site(
+        self, user_input: dict[str, str] | None = None
+    ) -> FlowResult:
         """Step to select site."""
         self._errors = {}
 
         assert self._sites is not None
+        assert self._api_token is not None
 
         api_token = self._api_token
         if user_input is not None: