Appdaemon sensor customization

Is it posisble to have a binary sensor created by app daemon using set_state to illuminate correctly in the interface. At the moment I can create sensors, but even thought the history and auomations recognize on/off state changes correctly, the little icon will never glow yellow :frowning:

I assume customize ymal, is loaded before the sensors exits… I also tried device_class as an attriubte but that does nothing.