diff --git a/custom_components/deltadore_tydom/ha_entities.py b/custom_components/deltadore_tydom/ha_entities.py
index d3483c5..b75782f 100644
--- a/custom_components/deltadore_tydom/ha_entities.py
+++ b/custom_components/deltadore_tydom/ha_entities.py
@@ -34,6 +34,7 @@
 from homeassistant.components.light import LightEntity
 from homeassistant.components.lock import LockEntity
 from homeassistant.components.update import UpdateEntity, UpdateEntityFeature, UpdateDeviceClass
+from homeassistant.components.alarm_control_panel import AlarmControlPanelEntity, CodeFormat
 
 from .tydom.tydom_devices import (
     Tydom,