Code review: use correct flow handler type in cast

home-assistant · Dec 28, 2024 · 24b1d88 · 24b1d88
1 parent 9e3964f
commit 24b1d88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/homeassistant/components/generic/config_flow.py b/homeassistant/components/generic/config_flow.py
@@ -578,14 +578,15 @@ async def ws_start_preview(
     _LOGGER.debug("Generating websocket handler for generic camera preview")
 
     flow_id = msg["flow_id"]
+    flow: GenericIPCamConfigFlow | GenericOptionsFlowHandler
     if msg.get("flow_type", "config_flow") == "config_flow":
         flow = cast(
             GenericIPCamConfigFlow,
             hass.config_entries.flow._progress.get(flow_id),  # noqa: SLF001
         )
     else:  # (flow type == "options flow")
         flow = cast(
-            GenericIPCamConfigFlow,
+            GenericOptionsFlowHandler,
             hass.config_entries.options._progress.get(flow_id),  # noqa: SLF001
         )
     user_input = flow.preview_cam