diff --git a/custom_components/network_scanner/__init__.py b/custom_components/network_scanner/__init__.py
index 4f6daa3..2f46e98 100644
--- a/custom_components/network_scanner/__init__.py
+++ b/custom_components/network_scanner/__init__.py
@@ -9,9 +9,7 @@ async def async_setup(hass, config):
 
 async def async_setup_entry(hass, config_entry):
     """Set up Network Scanner from a config entry."""
-    hass.async_create_task(
-        hass.config_entries.async_forward_entry_setup(config_entry, "sensor")
-    )
+    await hass.config_entries.async_forward_entry_setup(config_entry, ["sensor"])
     return True
 
 async def async_unload_entry(hass, config_entry):