Xiaomi Mi Smart LED Ceiling Light issue

Hello!

I have Xiaomi Mi Smart LED Ceiling Light 350 mm based on Yeelight, model MJXDD002:

I tried to connect it to the HA via Xiaomi Miot Home. Everything works well but after connection, I found some interesting issue.

In the Mi Home app this light has Wall Switch configuration.

Before connection to HA this light changed sequences after every “Turn Off/Turn on” action of wall switch.

But after connection to HA this configuration started working very strangely. To change sequences I need to do twice “Turn Off/Turn on” actions, because on the first “Turn on” the light is still off

How to return the behavior of the lamp on the wall switch to the previous mode?

Also, what is better to use? Xiaomi Miot Home or GitHub - XiaoMi/ha_xiaomi_home: Xiaomi Home Integration for Home Assistant?