diff --git a/custom_components/ecocito/__init__.py b/custom_components/ecocito/__init__.py
index 8336538..1fa729c 100644
--- a/custom_components/ecocito/__init__.py
+++ b/custom_components/ecocito/__init__.py
@@ -19,8 +19,6 @@ from .const import (
 )
 from .coordinator import (
     CollectionDataUpdateCoordinator,
-    # GarbageCollectionsDataUpdateCoordinator,
-    # RecyclingCollectionsDataUpdateCoordinator,
     WasteDepotVisitsDataUpdateCoordinator,
 )
 
diff --git a/custom_components/ecocito/config_flow.py b/custom_components/ecocito/config_flow.py
index bac8804..6c2bd6f 100644
--- a/custom_components/ecocito/config_flow.py
+++ b/custom_components/ecocito/config_flow.py
@@ -67,8 +67,8 @@ class EcocitoConfigFlow(ConfigFlow, domain=DOMAIN):
     @staticmethod
     @callback
     def async_get_options_flow(
-        config_entry: ConfigEntry
+        entry: ConfigEntry
     ) -> OptionsFlow:
         """Return the options flow."""
         from .options_flow import EcocitoOptionsFlowHandler
-        return EcocitoOptionsFlowHandler(config_entry)
+        return EcocitoOptionsFlowHandler(entry)
diff --git a/custom_components/ecocito/options_flow.py b/custom_components/ecocito/options_flow.py
index b32bd90..bdda0be 100644
--- a/custom_components/ecocito/options_flow.py
+++ b/custom_components/ecocito/options_flow.py
@@ -5,7 +5,7 @@ from __future__ import annotations
 from typing import Any
 
 import voluptuous as vol
-from homeassistant import ConfigFlowResult, config_entries
+from homeassistant import config_entries
 from homeassistant.const import CONF_DOMAIN, CONF_PASSWORD, CONF_USERNAME
 from homeassistant.helpers import selector
 
@@ -46,9 +46,9 @@ def build_schema(type_mapping: dict[int, str], current: dict[str, Any]) -> vol.S
 class EcocitoOptionsFlowHandler(config_entries.OptionsFlow):
     """Handle an option flow for ecocito."""
 
-    def __init__(self, config_entry: config_entries.ConfigEntry) -> None:
+    def __init__(self, entry: config_entries.ConfigEntry) -> None:
         """Init."""
-        self._entry = config_entry
+        self._entry = entry
 
 
     async def get_type_mapping(self) -> dict[int, str]:
@@ -82,7 +82,7 @@ class EcocitoOptionsFlowHandler(config_entries.OptionsFlow):
     async def async_step_init(
             self,
             user_input: dict[str, Any] | None = None
-        ) -> ConfigFlowResult:
+        ) -> config_entries.ConfigFlowResult:
         """Display configuration menu."""
         errors: dict[str, str] = {}
         if user_input is not None: