Yeelight toggle problem in floor plan

Configuration
1

Hello,

I have got problem with yeelight and floor plan.

In states i can switch on/off as many times I want and mostly it works, the called service is turn on / turn off.
But in floor plan (i coppied config) i use service toggle… and when you turn on the yeelight it becomes unavailable after turning on. nothing helps, I need to manually turn off light and then turn on.
config in floorplan.yaml:

    - name: Lights
      entities:
         - light.yeelight1
         - light.yeelight2
         - light.yeelight3
         - light.yeelight4
         - light.gateway
      states:
        - state: 'on'
          class: 'light-on'
        - state: 'off'
          class: 'light-off'
      action:
        service: toggle

is there a possibility to change service toggle to work as in state card (on/off)?