ESPhome deepsleep prevent

Hello,

I have got problem with preventing deep sleep using ESPhome ( I would like prevent deep sleep for OTA update)

My code in ESPhome:

mqtt:
  broker: 192.168.1.99
  port: 1883
  username: mqtt
  password: pass
  on_message:
    topic: balcony_sht31_wemos_deepsleep/ota_mode
    payload: 'ON'
    then:
      deep_sleep.prevent: deep_sleep_sht31

deep_sleep:
 id: deep_sleep_sht31
 run_duration: 35s
 sleep_duration: 2min

Using Mqtt.publish I send message:

topic: balcony_sht31_wemos_deepsleep/ota_mode
payload: ON
retain: true

When I publish the message for preventing the deepsleep while the node is sleeping, it doesn’t get the message when it wakes up. It only works when msg is publish while node is awake.

Do you have idea how to solve this problem?

did you solve this? automation?