diff --git a/custom_components/bosch_shc/device_trigger.py b/custom_components/bosch_shc/device_trigger.py
index 7bb94ad..19414da 100644
--- a/custom_components/bosch_shc/device_trigger.py
+++ b/custom_components/bosch_shc/device_trigger.py
@@ -4,7 +4,7 @@
 
 import voluptuous as vol
 from boschshcpy import SHCDevice, SHCSession
-from homeassistant.components.automation import TriggerActionType
+from homeassistant.helpers.trigger import TriggerActionType
 from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA
 from homeassistant.components.device_automation.exceptions import (
     InvalidDeviceAutomationConfig,