diff --git a/homeassistant/components/websocket_api/commands.py b/homeassistant/components/websocket_api/commands.py
index 35bd6b1cd7c..aad053e033c 100644
--- a/homeassistant/components/websocket_api/commands.py
+++ b/homeassistant/components/websocket_api/commands.py
@@ -942,7 +942,7 @@ async def handle_integration_wait(
 ) -> None:
     """Handle wait for integration command."""
 
-    domain = msg["domain"]
+    domain: str = msg["domain"]
     setup_done = hass.data.get(DATA_SETUP_DONE, {})
 
     # Wait for the integration to be set up