diff --git a/custom_components/smartthings/number.py b/custom_components/smartthings/number.py
index 9e81a86..9f93d39 100644
--- a/custom_components/smartthings/number.py
+++ b/custom_components/smartthings/number.py
@@ -9,7 +9,7 @@
 from pysmartthings import Attribute, Capability
 from pysmartthings.device import DeviceEntity
 
-from homeassistant.components.number import NumberEntity, MODE_AUTO
+from homeassistant.components.number import NumberEntity, NumberMode
 
 from . import SmartThingsEntity
 from .const import DATA_BROKERS, DOMAIN
@@ -34,7 +34,7 @@
             0,
             100,
             1,
-            MODE_AUTO,
+            NumberMode.AUTO,
         )
     ],
 }