diff --git a/custom_components/watersmart/__init__.py b/custom_components/watersmart/__init__.py
index 8b95502..77e0cd1 100644
--- a/custom_components/watersmart/__init__.py
+++ b/custom_components/watersmart/__init__.py
@@ -5,6 +5,7 @@
 from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME, Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.typing import ConfigType
 
 from .client import WaterSmartClient
@@ -14,6 +15,7 @@
 from .types import WaterSmartConfigEntry, WaterSmartData
 
 PLATFORMS: list[Platform] = [Platform.SENSOR]
+CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
 
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: