diff --git a/homeassistant/components/xiaomi_aqara/cover.py b/homeassistant/components/xiaomi_aqara/cover.py index f4bf1f269b5cfc..cd9190dca351f6 100644 --- a/homeassistant/components/xiaomi_aqara/cover.py +++ b/homeassistant/components/xiaomi_aqara/cover.py @@ -17,8 +17,12 @@ def setup_platform(hass, config, add_entities, discovery_info=None): for device in gateway.devices['cover']: model = device['model'] if model == 'curtain': + if 'proto' not in device or int(device['proto'][0:1]) == 1: + data_key = 'status' + else: + data_key = 'curtain_status' devices.append(XiaomiGenericCover(device, "Curtain", - 'status', gateway)) + data_key, gateway)) add_entities(devices)