diff --git a/custom_components/tedee/__init__.py b/custom_components/tedee/__init__.py
index 7a9f759..20521f2 100644
--- a/custom_components/tedee/__init__.py
+++ b/custom_components/tedee/__init__.py
@@ -3,6 +3,7 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_ACCESS_TOKEN
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntry
 from pytedee_async import TedeeClient
 
 from .const import DOMAIN
@@ -49,7 +50,11 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
 
     if unload_ok:
         hass.data[DOMAIN].pop(entry.entry_id)
-        hass.data[DOMAIN] = {}
 
     return unload_ok
-        
\ No newline at end of file
+
+async def async_remove_config_entry_device(
+    hass: HomeAssistant, config_entry: ConfigEntry, device_entry: DeviceEntry
+) -> bool:
+    """Remove a config entry from a device."""
+    return True
\ No newline at end of file
diff --git a/custom_components/tedee/lock.py b/custom_components/tedee/lock.py
index aa3b0dd..cb60307 100644
--- a/custom_components/tedee/lock.py
+++ b/custom_components/tedee/lock.py
@@ -1,6 +1,6 @@
 import logging
 
-from homeassistant.components.lock import SUPPORT_OPEN, LockEntity
+from homeassistant.components.lock import LockEntity, LockEntityFeature
 from homeassistant.const import (ATTR_BATTERY_CHARGING, ATTR_BATTERY_LEVEL,
                                  ATTR_ID)
 from homeassistant.core import callback
@@ -127,7 +127,7 @@ def __init__(self, lock, coordinator, entry):
 
     @property
     def supported_features(self):
-        return SUPPORT_OPEN
+        return LockEntityFeature.OPEN
 
 
     async def async_unlock(self, **kwargs):
diff --git a/custom_components/tedee/manifest.json b/custom_components/tedee/manifest.json
index 9f9723c..f4c34f0 100644
--- a/custom_components/tedee/manifest.json
+++ b/custom_components/tedee/manifest.json
@@ -6,7 +6,7 @@
   "dependencies": [],
   "codeowners": ["@patrickhilker", "@joerg65", "@zweckj"],
   "requirements": ["pytedee-async==0.1.4"],
-  "version": "2023.7.4",
+  "version": "2024.1.1",
   "config_flow": true,
   "iot_class": "cloud_polling"
 }