diff --git a/components/jiecang_desk_controller/__init__.py b/components/jiecang_desk_controller/__init__.py
index faa9915..d9afd09 100644
--- a/components/jiecang_desk_controller/__init__.py
+++ b/components/jiecang_desk_controller/__init__.py
@@ -26,13 +26,8 @@ JiecangDeskNumber = jiecang_desk_controller_ns.class_(
     'JiecangDeskNumber', number.Number, cg.Component
 )
 
-# -------------------------------------------------------------------
-# Robust schema compatibility (ESPHome 2025.11 ↔ 2025.12+)
-# -------------------------------------------------------------------
-BUTTON_SCHEMA = getattr(button, "_BUTTON_SCHEMA", button.BUTTON_SCHEMA)
-NUMBER_SCHEMA = getattr(number, "_NUMBER_SCHEMA", number.NUMBER_SCHEMA)
-
-# -------------------------------------------------------------------
+NUMBER_SCHEMA = number._NUMBER_SCHEMA
+BUTTON_SCHEMA = button._BUTTON_SCHEMA
 
 CONF_SENSORS = "sensors"
 CONF_BUTTONS = "buttons"