diff --git a/custom_components/wundasmart/strings.json b/custom_components/wundasmart/strings.json
index 0de5104..2a4b622 100644
--- a/custom_components/wundasmart/strings.json
+++ b/custom_components/wundasmart/strings.json
@@ -38,22 +38,16 @@
           "auto_on": "On (Auto)",
           "auto_off": "Off (Auto)",
           "boost_on": "On",
-          "boost_off": "Off"
-        },
-        "state_attributes": {
-          "operation_mode": {
-            "state": {
-              "auto": "Auto",
-              "boost_30": "Boost (30 mins)",
-              "boost_60": "Boost (1 hour)",
-              "boost_90": "Boost (1.5 hours)",
-              "boost_120": "Boost (2 hours)",
-              "off_30": "Off (30 mins)",
-              "off_60": "Off (1 hour)",
-              "off_90": "Off (1.5 hours)",
-              "off_120": "Off (2 hours)"
-            }
-          }
+          "boost_off": "Off",
+          "auto": "Auto",
+          "boost_30": "Boost (30 mins)",
+          "boost_60": "Boost (1 hour)",
+          "boost_90": "Boost (1.5 hours)",
+          "boost_120": "Boost (2 hours)",
+          "off_30": "Off (30 mins)",
+          "off_60": "Off (1 hour)",
+          "off_90": "Off (1.5 hours)",
+          "off_120": "Off (2 hours)"
         }
       }
     }
diff --git a/custom_components/wundasmart/translations/en.json b/custom_components/wundasmart/translations/en.json
index dd1c9d5..1f4d38e 100644
--- a/custom_components/wundasmart/translations/en.json
+++ b/custom_components/wundasmart/translations/en.json
@@ -38,22 +38,16 @@
                     "auto_on": "On (Auto)",
                     "auto_off": "Off (Auto)",
                     "boost_on": "On",
-                    "boost_off": "Off"
-                },
-                "state_attributes": {
-                    "operation_mode": {
-                        "state": {
-                            "auto": "Auto",
-                            "boost_30": "Boost (30 mins)",
-                            "boost_60": "Boost (1 hour)",
-                            "boost_90": "Boost (1.5 hours)",
-                            "boost_120": "Boost (2 hours)",
-                            "off_30": "Off (30 mins)",
-                            "off_60": "Off (1 hour)",
-                            "off_90": "Off (1.5 hours)",
-                            "off_120": "Off (2 hours)"
-                        }
-                    }
+                    "boost_off": "Off",
+                    "auto": "Auto",
+                    "boost_30": "Boost (30 mins)",
+                    "boost_60": "Boost (1 hour)",
+                    "boost_90": "Boost (1.5 hours)",
+                    "boost_120": "Boost (2 hours)",
+                    "off_30": "Off (30 mins)",
+                    "off_60": "Off (1 hour)",
+                    "off_90": "Off (1.5 hours)",
+                    "off_120": "Off (2 hours)"
                 }
             }
         }