diff --git a/components/jiecang_desk_controller/__init__.py b/components/jiecang_desk_controller/__init__.py
index f0ec16d..7e7e395 100644
--- a/components/jiecang_desk_controller/__init__.py
+++ b/components/jiecang_desk_controller/__init__.py
@@ -96,12 +96,12 @@ CONFIG_SCHEMA = cv.COMPONENT_SCHEMA.extend({
         ),
     }),
     cv.Optional(CONF_NUMBERS): cv.Schema({
-        cv.Optional(CONF_HEIGHT): number.NUMBER_SCHEMA.extend({
+        cv.Optional(CONF_HEIGHT): number._NUMBER_SCHEMA.extend({
             cv.GenerateID(): cv.declare_id(JiecangDeskNumber),
             cv.Optional(CONF_ACCURACY_DECIMALS, default=1): cv.int_,
             cv.Optional(CONF_UNIT_OF_MEASUREMENT, default=UNIT_CENTIMETER): cv.string,
         }),
-        cv.Optional(CONF_HEIGHT_PCT): number.NUMBER_SCHEMA.extend({
+        cv.Optional(CONF_HEIGHT_PCT): number._NUMBER_SCHEMA.extend({
             cv.GenerateID(): cv.declare_id(JiecangDeskNumber),
             cv.Optional(CONF_ACCURACY_DECIMALS, default=1): cv.int_,
             cv.Optional(CONF_UNIT_OF_MEASUREMENT, default=UNIT_PERCENT): cv.string,