diff --git a/custom_components/ge_home/devices/fridge.py b/custom_components/ge_home/devices/fridge.py
index 7c7aac8..5f5dd8c 100644
--- a/custom_components/ge_home/devices/fridge.py
+++ b/custom_components/ge_home/devices/fridge.py
@@ -1,4 +1,4 @@
-from homeassistant.components.binary_sensor import DEVICE_CLASS_PROBLEM
+from homeassistant.components.binary_sensor import BinarySensorDeviceClass
 from homeassistant.components.sensor import SensorDeviceClass
 import logging
 from typing import List
@@ -130,7 +130,7 @@ class FridgeApi(ApplianceApi):
                 GeErdPropertySensor(self, ErdCode.HOT_WATER_STATUS, "status", icon_override="mdi:information-outline"),
                 GeErdPropertySensor(self, ErdCode.HOT_WATER_STATUS, "time_until_ready", icon_override="mdi:timer-outline"),
                 GeErdPropertySensor(self, ErdCode.HOT_WATER_STATUS, "current_temp", device_class_override=SensorDeviceClass.TEMPERATURE, data_type_override=ErdDataType.INT),
-                GeErdPropertyBinarySensor(self, ErdCode.HOT_WATER_STATUS, "faulted", device_class_override=DEVICE_CLASS_PROBLEM),
+                GeErdPropertyBinarySensor(self, ErdCode.HOT_WATER_STATUS, "faulted", device_class_override=BinarySensorDeviceClass.PROBLEM),
                 GeDispenser(self)
             ])
 
diff --git a/custom_components/ge_home/entities/common/ge_climate.py b/custom_components/ge_home/entities/common/ge_climate.py
index c3c4583..e2f40dc 100644
--- a/custom_components/ge_home/entities/common/ge_climate.py
+++ b/custom_components/ge_home/entities/common/ge_climate.py
@@ -17,7 +17,7 @@ from .options_converter import OptionsConverter
 _LOGGER = logging.getLogger(__name__)
 
 #by default, we'll support target temp and fan mode (derived classes can override)
-GE_CLIMATE_SUPPORT = WaterHeaterEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.FAN_MODE
+GE_CLIMATE_SUPPORT = ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.FAN_MODE
 
 class GeClimate(GeEntity, ClimateEntity):
     """GE Climate Base Entity (Window AC, Portable AC, etc)"""
diff --git a/custom_components/ge_home/manifest.json b/custom_components/ge_home/manifest.json
index 33dd56f..88350da 100644
--- a/custom_components/ge_home/manifest.json
+++ b/custom_components/ge_home/manifest.json
@@ -7,5 +7,5 @@
   "documentation": "https://github.com/simbaja/ha_gehome",
   "requirements": ["gehomesdk==0.5.26","magicattr==0.1.6","slixmpp==1.8.3"],
   "codeowners": ["@simbaja"],	
-  "version": "0.6.9"
+  "version": "0.6.10"
 }