diff --git a/ariston/const.py b/ariston/const.py
index 13f9b09..18c47af 100644
--- a/ariston/const.py
+++ b/ariston/const.py
@@ -229,9 +229,9 @@ class NuosSplitOperativeMode(WaterHeaterMode):
     """NuosSplit operative mode enum"""
 
     GREEN = 0
-    COMFORT = 1
     FAST = 2
     IMEMORY = 3
+    GREEN_PLUS = 6
 
 
 @unique
diff --git a/ariston/nuos_split_device.py b/ariston/nuos_split_device.py
index bfc6fe7..e53f08c 100644
--- a/ariston/nuos_split_device.py
+++ b/ariston/nuos_split_device.py
@@ -175,7 +175,7 @@ async def async_set_water_heater_reduced_temperature(self, temperature: float):
     def set_water_heater_operation_mode(self, operation_mode: str):
         """Set water heater operation mode"""
         self.api.set_nuos_mode(self.gw, NuosSplitOperativeMode[operation_mode])
-        self.data[NuosSplitProperties.MODE] = NuosSplitOperativeMode[
+        self.data[NuosSplitProperties.OP_MODE] = NuosSplitOperativeMode[
             operation_mode
         ].value
 
@@ -184,7 +184,7 @@ async def async_set_water_heater_operation_mode(self, operation_mode: str):
         await self.api.async_set_nuos_mode(
             self.gw, NuosSplitOperativeMode[operation_mode]
         )
-        self.data[NuosSplitProperties.MODE] = NuosSplitOperativeMode[
+        self.data[NuosSplitProperties.OP_MODE] = NuosSplitOperativeMode[
             operation_mode
         ].value