diff --git a/homeassistant/helpers/config_entry_flow.py b/homeassistant/helpers/config_entry_flow.py
index 1a29f8a1752..05365b85645 100644
--- a/homeassistant/helpers/config_entry_flow.py
+++ b/homeassistant/helpers/config_entry_flow.py
@@ -9,7 +9,7 @@ from homeassistant.core import HomeAssistant
 from homeassistant.data_entry_flow import FlowResult
 from homeassistant.helpers.typing import UNDEFINED, DiscoveryInfoType, UndefinedType
 
-DiscoveryFunctionType = Callable[[], Union[Awaitable[bool], bool]]
+DiscoveryFunctionType = Callable[[HomeAssistant], Union[Awaitable[bool], bool]]
 
 _LOGGER = logging.getLogger(__name__)