Consistant use of conn and plm

homeassistant/components/insteon_plm.py

@@ -84,6 +84,7 @@ def async_setup(hass, config):
     _LOGGER.info('Config dir %s', hass.config.config_dir)
     _LOGGER.info("Looking for PLM on %s", port)
     conn = yield from insteonplm.Connection.create(device=port, loop=hass.loop, workdir=hass.config.config_dir)
+    plm = conn.protocol
 
     for device_override in overrides:
         #
@@ -102,11 +103,11 @@ def async_setup(hass, config):
             plm.protocol.devices.add_override(
                 device_override['address'], 'product_key', device_override['product_key'])
 
-    hass.data['insteon_plm'] = conn.protocol
+    hass.data['insteon_plm'] = plm
 
     hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, plm.close)
 
-    plm.protocol.devices.add_device_callback(async_plm_new_device)
+    plm.devices.add_device_callback(async_plm_new_device)
 
     return True