diff --git a/lib/models/requests/set_state_request.dart b/lib/models/requests/set_state_request.dart
index f504cad..f6305c5 100644
--- a/lib/models/requests/set_state_request.dart
+++ b/lib/models/requests/set_state_request.dart
@@ -1,5 +1,6 @@
 import "package:freezed_annotation/freezed_annotation.dart";
 import "package:nexus/models/content/content.dart";
+import "package:nexus/models/ms_duration.dart";
 part "set_state_request.freezed.dart";
 part "set_state_request.g.dart";
 
@@ -10,7 +11,7 @@ abstract class SetStateRequest with _$SetStateRequest {
     required String type,
     required String stateKey,
     required Content content,
-    @Default(0) int delayMs,
+    @JsonKey(name: "delay_ms") @MSDuration() @Default(null) Duration? delay,
   }) = _SetStateRequest;
 
   factory SetStateRequest.fromJson(Map<String, Object?> json) =>