@@ -103,19 +103,28 @@ def __init__(self, hass: HomeAssistant, yidcal_helper) -> None:
103103 self .entity_id = "sensor.yidcal_tehilim_daily_pupa"
104104 self ._state : str | None = None
105105
106+ async def async_added_to_hass (self ) -> None :
107+ await super ().async_added_to_hass ()
108+
106109 # initial compute
107110 self .update ()
108111
109112 # refresh after midnight (civil)
110- async_track_time_change (
111- hass , self ._handle_midnight , hour = 0 , minute = 0 , second = 1
113+ self ._register_listener (
114+ async_track_time_change (
115+ self .hass , self ._handle_midnight , hour = 0 , minute = 0 , second = 1
116+ )
112117 )
113118
114- async_track_state_change_event (
115- hass , "binary_sensor.yidcal_no_melucha" , self ._handle_midnight
119+ self ._register_listener (
120+ async_track_state_change_event (
121+ self .hass , "binary_sensor.yidcal_no_melucha" , self ._handle_midnight
122+ )
116123 )
117- async_track_state_change_event (
118- hass , "sensor.yidcal_holiday" , self ._handle_midnight
124+ self ._register_listener (
125+ async_track_state_change_event (
126+ self .hass , "sensor.yidcal_holiday" , self ._handle_midnight
127+ )
119128 )
120129
121130 async def _handle_midnight (self , * _ ):
0 commit comments