Home App Switch turns off automatically

Hi all,

I successfully added button to Home APP via HA and broadlink virtual remote.
However the switch is not functioning well in the APP, here is the problem:

click once → turn ‘on’ the light successfully.
3 second later (didn’t click anything) → APP button changed to ‘off’ but the light is still on.
click once → APP button changed to ‘on’ again, but the light is off now.

I wonder if there is any way to put both on and off in one button (one click on-> one click off) can work well.

Any help would be great. Thank you