diff --git a/__init__.py b/__init__.py
index 67905af..072a526 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,14 +1,11 @@
 """The SNCF integration."""
 from __future__ import annotations
-
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
-
 from .const import DOMAIN
 
-# TODO List the platforms that you want to support.
 # For your initial PR, limit it to 1 platform.
-PLATFORMS: list[str] = ["light"]
+PLATFORMS: list[str] = ["sensor"]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
diff --git a/config_flow.py b/config_flow.py
index 45fd302..0126722 100644
--- a/config_flow.py
+++ b/config_flow.py
@@ -8,7 +8,7 @@ from homeassistant import config_entries
 from homeassistant.core import HomeAssistant
 from homeassistant.data_entry_flow import FlowResult
 from homeassistant.exceptions import HomeAssistantError
-from sncf import SNCF
+from .sncf import SNCF
 from .const import DOMAIN
 
 _LOGGER = logging.getLogger(__name__)