diff --git a/custom_components/ecowater_softener/__init__.py b/custom_components/ecowater_softener/__init__.py
index 07be0f2..c7ca256 100644
--- a/custom_components/ecowater_softener/__init__.py
+++ b/custom_components/ecowater_softener/__init__.py
@@ -22,9 +22,8 @@ async def async_setup_entry(
     hass.data[DOMAIN][entry.entry_id] = hass_data
 
     # Forward the setup to the sensor platform.
-    hass.async_create_task(
-        hass.config_entries.async_forward_entry_setup(entry, "sensor")
-    )
+    await hass.config_entries.async_forward_entry_setups(entry, ["sensor"])
+    
     return True
 
 
@@ -51,4 +50,4 @@ async def async_unload_entry(
     if unload_ok:
         hass.data[DOMAIN].pop(entry.entry_id)
 
-    return unload_ok
\ No newline at end of file
+    return unload_ok