diff --git a/drivers/nuos/device.ts b/drivers/nuos/device.ts
index 49276dd..b94b569 100644
--- a/drivers/nuos/device.ts
+++ b/drivers/nuos/device.ts
@@ -152,10 +152,10 @@ class NuosDevice extends Device {
   }
 
   readonly #diff = new Map<
-    keyof Capabilities,
+    keyof SetCapabilities,
     {
-      initialValue: Capabilities[keyof Capabilities]
-      value: Capabilities[keyof Capabilities]
+      initialValue: SetCapabilities[keyof SetCapabilities]
+      value: SetCapabilities[keyof SetCapabilities]
     }
   >()