diff --git a/tests/components/template/common-base.yaml b/tests/components/template/common-base.yaml
index 119afb22aa..352b65c0f6 100644
--- a/tests/components/template/common-base.yaml
+++ b/tests/components/template/common-base.yaml
@@ -414,7 +414,7 @@ water_heater:
     current_temperature: !lambda "return 42.0f;"
     target_temperature: !lambda "return 60.0f;"
     mode: !lambda "return water_heater::WATER_HEATER_MODE_ECO;"
-    on: !lambda "return true;"
+    "on": !lambda "return true;"
     away: !lambda "return false;"
     supported_modes:
       - "OFF"
@@ -428,11 +428,11 @@ water_heater:
       - logger.log: "set_action"
       - water_heater.template.publish:
           id: template_water_heater
-          on: false
+          "on": false
           away: true
       - water_heater.template.publish:
           id: template_water_heater
-          on: !lambda "return true;"
+          "on": !lambda "return true;"
           away: !lambda "return false;"
 
 datetime:
diff --git a/tests/integration/fixtures/water_heater_template.yaml b/tests/integration/fixtures/water_heater_template.yaml
index f04a2a86b6..3e2503209e 100644
--- a/tests/integration/fixtures/water_heater_template.yaml
+++ b/tests/integration/fixtures/water_heater_template.yaml
@@ -11,7 +11,7 @@ water_heater:
     optimistic: true
     current_temperature: !lambda "return 45.0f;"
     target_temperature: !lambda "return 60.0f;"
-    on: !lambda "return true;"
+    "on": !lambda "return true;"
     away: !lambda "return false;"
     # Note: No mode lambda - we want optimistic mode changes to stick
     # A mode lambda would override mode changes in loop()