fix #357

KoljaWindeler · Dec 29, 2024 · fcd4f94 · fcd4f94
1 parent 027e49b
commit fcd4f94
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/custom_components/ytube_music_player/__init__.py b/custom_components/ytube_music_player/__init__.py
@@ -20,10 +20,7 @@ async def async_setup_entry(hass, config_entry):
 		config_entry.add_update_listener(async_update_options)
 
 	# Add entities to HA
-	for platform in PLATFORMS:
-		hass.async_create_task(
-			hass.config_entries.async_forward_entry_setup(config_entry, platform)
-		)
+	await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
 	return True
 
 
@@ -44,4 +41,4 @@ async def async_update_options(hass, config_entry):
 	_LOGGER.debug("Config updated,reload the entities.")
 	for platform in PLATFORMS:
 		await hass.config_entries.async_forward_entry_unload(config_entry, platform)
-		hass.async_create_task(hass.config_entries.async_forward_entry_setup(config_entry, platform))
+	await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)