diff --git a/lib/pages/device_managment/schedule_device/schedule_widgets/schedule_table.dart b/lib/pages/device_managment/schedule_device/schedule_widgets/schedule_table.dart
index b23e48df..21f404ff 100644
--- a/lib/pages/device_managment/schedule_device/schedule_widgets/schedule_table.dart
+++ b/lib/pages/device_managment/schedule_device/schedule_widgets/schedule_table.dart
@@ -195,10 +195,10 @@ class _ScheduleTableView extends StatelessWidget {
             child: Text(_getSelectedDays(
                 ScheduleModel.parseSelectedDays(schedule.days)))),
         Center(child: Text(formatIsoStringToTime(schedule.time, context))),
-        schedule.category == 'CUR_2'
-            ? Center(
-                child: Text(schedule.function.value == true ? 'open' : 'close'))
-            : Center(child: Text(schedule.function.value ? 'On' : 'Off')),
+        if (schedule.category == 'CUR_2')
+          Center(child: Text(schedule.function.value))
+        else
+          Center(child: Text(schedule.function.value ? 'On' : 'Off')),
         Center(
           child: Wrap(
             runAlignment: WrapAlignment.center,
diff --git a/lib/pages/device_managment/water_heater/helper/add_schedule_dialog_helper.dart b/lib/pages/device_managment/water_heater/helper/add_schedule_dialog_helper.dart
index 389eac3f..f55b32ab 100644
--- a/lib/pages/device_managment/water_heater/helper/add_schedule_dialog_helper.dart
+++ b/lib/pages/device_managment/water_heater/helper/add_schedule_dialog_helper.dart
@@ -19,13 +19,19 @@ class ScheduleDialogHelper {
     bool isEdit = false,
     String? code,
   }) {
+    bool temp;
+    if (schedule?.category == 'CUR_2') {
+      temp = schedule!.function.value == 'open' ? true : false;
+    } else {
+      temp = schedule!.function.value;
+    }
     final initialTime = schedule != null
         ? _convertStringToTimeOfDay(schedule.time)
         : TimeOfDay.now();
     final initialDays = schedule != null
         ? _convertDaysStringToBooleans(schedule.days)
         : List.filled(7, false);
-    bool? functionOn = schedule?.function.value ?? true;
+    bool? functionOn = temp;
     TimeOfDay selectedTime = initialTime;
     List<bool> selectedDays = List.of(initialDays);