diff --git a/esphomeyaml/components/output/__init__.py b/esphomeyaml/components/output/__init__.py
index cb8dda9d12..22cfaffffa 100644
--- a/esphomeyaml/components/output/__init__.py
+++ b/esphomeyaml/components/output/__init__.py
@@ -21,7 +21,7 @@ FLOAT_OUTPUT_SCHEMA = BINARY_OUTPUT_SCHEMA.extend({
 output_ns = esphomelib_ns.namespace('output')
 
 
-def setup_output_platform(obj, config, skip_power_supply=False):
+def setup_output_platform_(obj, config, skip_power_supply=False):
     if CONF_INVERTED in config:
         add(obj.set_inverted(config[CONF_INVERTED]))
     if not skip_power_supply and CONF_POWER_SUPPLY in config:
@@ -33,4 +33,9 @@ def setup_output_platform(obj, config, skip_power_supply=False):
         add(obj.set_max_power(config[CONF_MAX_POWER]))
 
 
+def setup_output_platform(obj, config, skip_power_supply=False):
+    for _ in setup_output_platform_(obj, config, skip_power_supply):
+        yield
+
+
 BUILD_FLAGS = '-DUSE_OUTPUT'