diff --git a/pyworxcloud/utils/capability.py b/pyworxcloud/utils/capability.py index f60adfa..0554ede 100644 --- a/pyworxcloud/utils/capability.py +++ b/pyworxcloud/utils/capability.py @@ -39,7 +39,10 @@ def __init__(self, device_data: Any | None = None) -> int: try: if "sc" in device_data["last_status"]["payload"]["cfg"]: - if "ots" in device_data["last_status"]["payload"]["cfg"]["sc"]: + if ( + "ots" in device_data["last_status"]["payload"]["cfg"]["sc"] + or "once" in device_data["last_status"]["payload"]["cfg"]["sc"] + ): self.add(DeviceCapability.ONE_TIME_SCHEDULE) self.add(DeviceCapability.EDGE_CUT) if (