diff --git a/api/profile.php b/api/profile.php
index 2e3641d..1cbfc33 100644
--- a/api/profile.php
+++ b/api/profile.php
@@ -266,8 +266,8 @@ class profile extends cora\api {
 
           // Normalizing heating and cooling a bit.
           if(
-            $thermostat['system_type']['detected']['heat'] === 'compressor' ||
-            $thermostat['system_type']['detected']['heat'] === 'geothermal'
+            $thermostat['system_type2']['detected']['heat']['equipment'] === 'compressor' ||
+            $thermostat['system_type2']['detected']['heat']['equipment'] === 'geothermal'
           ) {
             if($row['compressor_mode'] === 'heat') {
               $row['heat_1'] = $row['compressor_1'];