diff --git a/custom_components/zcsmower/vacuum.py b/custom_components/zcsmower/vacuum.py index 2a12cfe..a27079c 100644 --- a/custom_components/zcsmower/vacuum.py +++ b/custom_components/zcsmower/vacuum.py @@ -8,9 +8,9 @@ from homeassistant.config_entries import ConfigEntry from homeassistant.components.vacuum import ( ATTR_STATUS, - VacuumActivity, StateVacuumEntity, StateVacuumEntityDescription, + VacuumActivity, VacuumEntityFeature, ) from homeassistant.helpers.entity import Entity