diff --git a/custom_components/fullykiosk/config_flow.py b/custom_components/fullykiosk/config_flow.py
index 31f77bf..729eb82 100644
--- a/custom_components/fullykiosk/config_flow.py
+++ b/custom_components/fullykiosk/config_flow.py
@@ -46,7 +46,7 @@ async def authenticate(self, username, password) -> bool:
 
 async def validate_input(hass: core.HomeAssistant, data):
     """Validate the user input allows us to connect."""
-    session = async_get_clientsession()
+    session = async_get_clientsession(hass)
     fully = FullyKiosk(session, data["host"], data["port"], data["password"])
 
     try: