diff --git a/homeassistant/components/demo/sensor.py b/homeassistant/components/demo/sensor.py
index 26689582fae..a1f7504762a 100644
--- a/homeassistant/components/demo/sensor.py
+++ b/homeassistant/components/demo/sensor.py
@@ -25,7 +25,6 @@ from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.event import async_track_time_interval
-from homeassistant.helpers.typing import StateType
 
 from . import DOMAIN
 
@@ -149,11 +148,11 @@ class DemoSensor(SensorEntity):
         self,
         unique_id: str,
         device_name: str | None,
-        state: StateType,
+        state: float | int | str | None,
         device_class: SensorDeviceClass,
         state_class: SensorStateClass | None,
         unit_of_measurement: str | None,
-        battery: StateType,
+        battery: int | None,
         options: list[str] | None = None,
         translation_key: str | None = None,
     ) -> None:
@@ -189,7 +188,7 @@ class DemoSumSensor(RestoreSensor):
         device_class: SensorDeviceClass,
         state_class: SensorStateClass | None,
         unit_of_measurement: str | None,
-        battery: StateType,
+        battery: int | None,
         suggested_entity_id: str,
     ) -> None:
         """Initialize the sensor."""