I might be late to the party, but I have just faced the same problem, and I couldn’t find this solution on the forums. Fortunately I could solve the issue by specifying simple-entity as the type of the entity object (I found this option here:
entities:
- entity: switch.wifi_plug_2_socket_1
type: simple-entity
This makes the switch show its state rather than the toggle:
