diff --git a/esphomeyaml/components/my9231.py b/esphomeyaml/components/my9231.py
index a029b606da..7f0b7b87cd 100644
--- a/esphomeyaml/components/my9231.py
+++ b/esphomeyaml/components/my9231.py
@@ -7,7 +7,7 @@ from esphomeyaml.const import (CONF_DATA_PIN, CONF_CLOCK_PIN, CONF_NUM_CHANNELS,
                                CONF_NUM_CHIPS, CONF_BIT_DEPTH, CONF_ID,
                                CONF_UPDATE_ON_BOOT)
 from esphomeyaml.helpers import (gpio_output_pin_expression, App, Pvariable,
-                                 add)
+                                 add, setup_component)
 
 MY9231OutputComponent = output.output_ns.namespace('MY9231OutputComponent')
 
@@ -23,7 +23,7 @@ MY9231_SCHEMA = vol.Schema({
     vol.Optional(CONF_BIT_DEPTH): vol.All(vol.Coerce(int),
                                           cv.one_of(8, 12, 14, 16)),
     vol.Optional(CONF_UPDATE_ON_BOOT): vol.Coerce(bool),
-})
+}).extend(cv.COMPONENT_SCHEMA.schema)
 
 CONFIG_SCHEMA = vol.All(cv.ensure_list, [MY9231_SCHEMA])
 
@@ -46,6 +46,7 @@ def to_code(config):
             add(my9231.set_bit_depth(conf[CONF_BIT_DEPTH]))
         if CONF_UPDATE_ON_BOOT in conf:
             add(my9231.set_update(conf[CONF_UPDATE_ON_BOOT]))
+        setup_component(my9231, conf)
 
 
 BUILD_FLAGS = '-DUSE_MY9231_OUTPUT'
diff --git a/esphomeyaml/components/output/my9231.py b/esphomeyaml/components/output/my9231.py
index 4aa9ad364e..31d5381075 100644
--- a/esphomeyaml/components/output/my9231.py
+++ b/esphomeyaml/components/output/my9231.py
@@ -4,7 +4,7 @@ import esphomeyaml.config_validation as cv
 from esphomeyaml.components import output
 from esphomeyaml.components.my9231 import MY9231OutputComponent
 from esphomeyaml.const import CONF_CHANNEL, CONF_ID, CONF_MY9231_ID, CONF_POWER_SUPPLY
-from esphomeyaml.helpers import Pvariable, get_variable
+from esphomeyaml.helpers import Pvariable, get_variable, setup_component
 
 DEPENDENCIES = ['my9231']
 
@@ -15,7 +15,7 @@ PLATFORM_SCHEMA = output.FLOAT_OUTPUT_PLATFORM_SCHEMA.extend({
     vol.Required(CONF_CHANNEL): vol.All(vol.Coerce(int),
                                         vol.Range(min=0, max=65535)),
     cv.GenerateID(CONF_MY9231_ID): cv.use_variable_id(MY9231OutputComponent),
-})
+}).extend(cv.COMPONENT_SCHEMA.schema)
 
 
 def to_code(config):
@@ -29,6 +29,7 @@ def to_code(config):
     rhs = my9231.create_channel(config[CONF_CHANNEL], power_supply)
     out = Pvariable(config[CONF_ID], rhs)
     output.setup_output_platform(out, config, skip_power_supply=True)
+    setup_component(out, config)
 
 
 BUILD_FLAGS = '-DUSE_MY9231_OUTPUT'