Mobile App Notification Action Received but Automation not Firing

I am trying to get an actionable notification working. As a test, I have it turning off a light, but nothing is happening. I can see the mobile_app_notification_action event being received using developer tools and can also see the Wait for Trigger in my Automation being triggered in the UI. So it seems like HA is definitely receiving the action key but the light doesn’t turn off. I’ve verified that it works if I run just the light turn off action. Here is my code:

alias: Yoga Tuesdays
description: ""
triggers:
  - at: "08:30:00"
    trigger: time
conditions:
  - condition: time
    weekday:
      - tue
actions:
  - data:
      title: Yoga Time!
      message: Ready to get bendy?
      data:
        importance: high
        actions:
          - action: YES_YOGA
            title: Yoga!
        notification_icon: mdi:yoga
    action: notify.samson
  - wait_for_trigger:
      - trigger: event
        event_type: mobile_app_notification_action
        event_data:
          action: YES_YOGA
    enabled: true
    continue_on_timeout: true
  - action: light.turn_off
    metadata: {}
    data: {}
    target:
      entity_id: light.living_room_double_plug_astronaut_light
mode: restart

Any idea what I’m missing?

This was very odd but I got it working/it was always working. For anyone encountering this issue, select ‘Run’ from the menu for the whole automation, not just the notify action. When the notification is produced by running the full automation, it works as expected. I don’t understand why this makes a difference, but it does not work if I try to test by running just the notify action.