diff --git a/lib/pages/device_managment/schedule_device/bloc/schedule_bloc.dart b/lib/pages/device_managment/schedule_device/bloc/schedule_bloc.dart
index fbf7ae64..0ec55e39 100644
--- a/lib/pages/device_managment/schedule_device/bloc/schedule_bloc.dart
+++ b/lib/pages/device_managment/schedule_device/bloc/schedule_bloc.dart
@@ -265,7 +265,7 @@ class ScheduleBloc extends Bloc<ScheduleEvent, ScheduleState> {
             category: event.category,
             deviceId: deviceId,
             time: getTimeStampWithoutSeconds(dateTime).toString(),
-            code: event.category,
+            code: event.code ?? event.category,
             value: event.functionOn,
             days: event.selectedDays);
         if (success) {
diff --git a/lib/pages/device_managment/schedule_device/bloc/schedule_event.dart b/lib/pages/device_managment/schedule_device/bloc/schedule_event.dart
index 0b9ec581..a28b8757 100644
--- a/lib/pages/device_managment/schedule_device/bloc/schedule_event.dart
+++ b/lib/pages/device_managment/schedule_device/bloc/schedule_event.dart
@@ -70,17 +70,19 @@ class ScheduleAddEvent extends ScheduleEvent {
   final String category;
   final String time;
   final List<String> selectedDays;
-  final bool functionOn;
+  final dynamic functionOn;
+  final String? code;
 
   const ScheduleAddEvent({
     required this.category,
     required this.time,
     required this.selectedDays,
     required this.functionOn,
+    required this.code,
   });
 
   @override
-  List<Object> get props => [category, time, selectedDays, functionOn];
+  List<Object?> get props => [category, time, selectedDays, functionOn, code];
 }
 
 class ScheduleEditEvent extends ScheduleEvent {