[BUG] light or switch?

[trytogetme85]

Environment

• ShellyForHASS release with the issue: 0.2.2
• Home Assistant Core release with the issue: 2021.5.5
• Operating environment Raspy

Describe the bug

I recently added 5 Shellys in my Lovelace (floorplan). After a restart some of the shellys are not shown online. I have to manualy change the integration from light.shellyxy to switch.shelly.xy..... after a restart of HA I have to to the opposite... see here after a restart only 4/5 shellys are shown correct, one shelly is "offline". I have to change the entity from switch. to light...

• type: image entity: light.shelly_shsw_1_e8db84d6557e style: left: 50% top: 50% width: 100% state_image: 'on': /local/images/floorplan/bad_100.png 'off': /local/images/floorplan/transparent.png

Screenshots

https://imgur.com/a/EKMp7Sz

https://imgur.com/XKjJRPc