diff --git a/esphome/components/sts3x/sensor.py b/esphome/components/sts3x/sensor.py
index 2715bdae39..6c16cd49fc 100644
--- a/esphome/components/sts3x/sensor.py
+++ b/esphome/components/sts3x/sensor.py
@@ -1,7 +1,7 @@
 import esphome.codegen as cg
 import esphome.config_validation as cv
 from esphome.components import i2c, sensor
-from esphome.const import CONF_ID, CONF_TEMPERATURE, ICON_THERMOMETER, UNIT_CELSIUS
+from esphome.const import CONF_ID, ICON_THERMOMETER, UNIT_CELSIUS
 
 DEPENDENCIES = ['i2c']
 
@@ -9,7 +9,7 @@ sts3x_ns = cg.esphome_ns.namespace('sts3x')
 
 STS3XComponent = sts3x_ns.class_('STS3XComponent', sensor.Sensor, cg.PollingComponent, i2c.I2CDevice)
 
-CONFIG_SCHEMA = sensor.sensor_schema(CONF_TEMPERATURE, ICON_THERMOMETER, 1).extend({
+CONFIG_SCHEMA = sensor.sensor_schema(UNIT_CELSIUS, ICON_THERMOMETER, 1).extend({
     cv.GenerateID(): cv.declare_id(STS3XComponent),
 }).extend(cv.polling_component_schema('60s')).extend(i2c.i2c_device_schema(0x4A))