diff --git a/tests/test_switch_ac.py b/tests/test_switch_ac.py
index 86082055..46641309 100644
--- a/tests/test_switch_ac.py
+++ b/tests/test_switch_ac.py
@@ -56,7 +56,7 @@ def find_my_entity(entity_id) -> ClimateEntity:
         assert entity.ac_mode is True
         assert entity.hvac_action is HVACAction.OFF
         assert entity.hvac_mode is HVACMode.OFF
-        assert entity.hvac_modes == [HVACMode.COOL, HVACMode.OFF]
+        assert entity.hvac_modes == [HVACMode.HEAT, HVACMode.COOL, HVACMode.OFF]
         assert entity.target_temperature == entity.max_temp
         assert entity.preset_modes == [
             PRESET_NONE,