diff --git a/custom_components/ssh/sensor.py b/custom_components/ssh/sensor.py
index 477389c..2fe8e28 100644
--- a/custom_components/ssh/sensor.py
+++ b/custom_components/ssh/sensor.py
@@ -41,8 +41,7 @@
     vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
 })
 
-@asyncio.coroutine
-def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
+async def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
 
     dev = []
     dev.append(SSHSensor(hass, config))