diff --git a/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml b/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml
index bad2704..75bfc76 100644
--- a/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml
+++ b/ESPHome/TX-Ultimate-Easy-ESPHome_core.yaml
@@ -162,7 +162,7 @@ select:
     icon: mdi:dip-switch
     on_value:
       then:
-        - lambda: set_gang_count(static_cast<uint8_t>(i)+1);
+        - lambda: tx_ultimate->set_gang_count(static_cast<uint8_t>(i)+1);
 
 text_sensor:
   - id: tx_fw_version
diff --git a/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_touch.yaml b/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_touch.yaml
index 8c9f090..4c3dfd4 100644
--- a/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_touch.yaml
+++ b/ESPHome/TX-Ultimate-Easy-ESPHome_core_hw_touch.yaml
@@ -262,7 +262,7 @@ script:
       - lambda: |-
           const uint8_t num_gangs = (sl_tx_model_gang->active_index().has_value()) ?
                                     (sl_tx_model_gang->active_index().value() + 1) : 0;
-          set_gang_count(num_gangs);
+          tx_ultimate->set_gang_count(num_gangs);
           bs_button_1_click_event->publish_state(false);
           bs_button_2_click_event->publish_state(false);
           bs_button_3_click_event->publish_state(false);