diff --git a/src/data/history.ts b/src/data/history.ts
index 0856e9eff3..73333cfd66 100644
--- a/src/data/history.ts
+++ b/src/data/history.ts
@@ -13,6 +13,7 @@ const LINE_ATTRIBUTES_TO_KEEP = [
   "target_temp_high",
   "hvac_action",
   "humidity",
+  "mode",
 ];
 
 export interface LineChartState {
diff --git a/src/util/hass-attributes-util.js b/src/util/hass-attributes-util.js
index 04f8ab2701..4eb17735a0 100644
--- a/src/util/hass-attributes-util.js
+++ b/src/util/hass-attributes-util.js
@@ -37,6 +37,7 @@ hassAttributeUtil.DOMAIN_DEVICE_CLASS = {
     "shutter",
     "window",
   ],
+  humidifier: ["dehumidifier", "humidifier"],
   sensor: [
     "battery",
     "humidity",