diff --git a/package/base.yaml b/package/base.yaml
index 64e3fa1..b053c6c 100644
--- a/package/base.yaml
+++ b/package/base.yaml
@@ -135,16 +135,16 @@ script:
       - script.execute: update_page
   - id: update_page
     then:
-      - lambda: |-
-          id(active_page_name) = id(page_names)[id(actual_page)];
-      # - sensor.template.publish:
-      #     id: active_page_id_ha
-      #     state: !lambda 'return id(actual_page);'
-      # - text_sensor.template.publish:
-      #     id: active_page_name_ha
-      #     state: !lambda 'return id(active_page_name);'
-      # - number.set:
-      #     id: menuselector
-      #     value: !lambda 'return id(actual_page);'
+      # - lambda: |-
+      #     id(active_page_name) = id(page_names)[id(actual_page)]; ## <<< This causes the button to reboot
+      - sensor.template.publish:
+          id: active_page_id_ha
+          state: !lambda 'return id(actual_page);'
+      - text_sensor.template.publish:
+          id: active_page_name_ha
+          state: !lambda 'return id(active_page_name);'
+      - number.set:
+          id: menuselector
+          value: !lambda 'return id(actual_page);'
       - script.execute: update_all