diff --git a/js/layer/load.js b/js/layer/load.js
index db13632..1bf5fe1 100644
--- a/js/layer/load.js
+++ b/js/layer/load.js
@@ -155,6 +155,10 @@ beestat.layer.load.prototype.decorate_ = function(parent) {
       beestat.setting('thermostat_id')
     ];
 
+    var ecobee_thermostat = beestat.cache.ecobee_thermostat[
+      thermostat.ecobee_thermostat_id
+    ];
+
     // Set the document title to include the thermostat name
     if (thermostat.name !== null && thermostat.name.trim() !== '') {
       document.title = 'beestat | ' + thermostat.name;
@@ -174,7 +178,8 @@ beestat.layer.load.prototype.decorate_ = function(parent) {
     }
 
     // Fix some other stuff for non-heat-pump.
-    if (beestat.thermostat.get_system_type(thermostat.thermostat_id, 'heat') !== 'compressor') {
+    if (ecobee_thermostat.settings.hasHeatPump === false) {
+    // if (beestat.thermostat.get_system_type(thermostat.thermostat_id, 'heat') !== 'compressor') {
       beestat.series.auxiliary_heat_1.name = beestat.series.sum_compressor_heat_1.name;
       beestat.series.auxiliary_heat_1.color = beestat.series.sum_compressor_heat_1.color;
       beestat.series.sum_auxiliary_heat_2.name = beestat.series.compressor_heat_2.name;