diff --git a/homeassistant/components/axis/config_flow.py b/homeassistant/components/axis/config_flow.py
index 1ce2f08c045..1fb9b9488fa 100644
--- a/homeassistant/components/axis/config_flow.py
+++ b/homeassistant/components/axis/config_flow.py
@@ -223,14 +223,10 @@ class AxisFlowHandler(config_entries.ConfigFlow, domain=AXIS_DOMAIN):
         return await self.async_step_user()
 
 
-class AxisOptionsFlowHandler(config_entries.OptionsFlow):
+class AxisOptionsFlowHandler(config_entries.OptionsFlowWithConfigEntry):
     """Handle Axis device options."""
 
-    def __init__(self, config_entry: ConfigEntry) -> None:
-        """Initialize Axis device options flow."""
-        self.config_entry = config_entry
-        self.options = dict(config_entry.options)
-        self.device: AxisNetworkDevice | None = None
+    device: AxisNetworkDevice
 
     async def async_step_init(
         self, user_input: dict[str, Any] | None = None
@@ -249,7 +245,6 @@ class AxisOptionsFlowHandler(config_entries.OptionsFlow):
 
         schema = {}
 
-        assert self.device
         vapix = self.device.api.vapix
 
         # Stream profiles