Sensor Template in Heading Badge Farbe dynamisch

Hey Leute,
ich bau gerade mein erstes Dashboard und habe hierfür einen Template Sensor erstellt, der mir die aktuelle Betriebsart meines Thermostates ausliest.

template:
  - sensor:
    - name: "Status Heizung Arbeitszimmer"
      unique_id: StHzgAz
      state: >-
          {% if is_state_attr('climate.heizung_arbeitszimmer','hvac_action', 'heating') -%}
            heating
          {% elif is_state_attr('climate.heizung_arbeitszimmer','hvac_action', 'idle') -%}
            idle
          {%- else -%}
            unknown
          {%- endif %}
      icon: >-
          {% if is_state_attr('climate.heizung_arbeitszimmer','hvac_action', 'heating') -%}
            mdi:radiator
          {% elif is_state_attr('climate.heizung_arbeitszimmer','hvac_action', 'idle') -%}
            mdi:radiator-disabled
          {%- else -%}
            mdi:help-rhombus-outline
          {%- endif %}

Das funktioniert soweit auch super.
Nun möchte ich diesen Sensor in einer Überschriften Karte als Badge anzeigen. Er soll aber die Farbe je nach Status (heating,idle) des Sensors wechseln.
Unbenannt
Ich bekomme es ums verrecken nicht hin. Habe gedacht einfach in den Einstellungen der Entität in der Karte also iwie so:

type: heading
icon: ""
heading: Arbeitszimmer
heading_style: title
badges:
  - type: entity
    show_state: true
    show_icon: true
    entity: sensor.status_heizung_arbeitszimmer
    color: >
      {% if is_state_attr('climate.heizung_arbeitszimmer','hvac_action',
      'heating') -%}
          deep-orange
      {% elif is_state_attr('climate.heizung_arbeitszimmer','hvac_action',
      'idle') -%}
          blue
      {%- else -%}
          red
      {%- endif %}

Leider Funktioniert das nicht… Jmd ne Idee? Ich komm noch nicht so ganz mit den ganzen unterschiedlichen Syntax/Schreibweisen klar.