Monitor_docker display different colours for different states

Hey,

I’ve installed the monitor_docker integration via HACS, everything went well, I can see the sensors I’ve selected and display them.

I’ve configured a badge to show me the state (so running, exited) of a container, which updates fine, however the colour does not change depending on the state, I’ve selected “Default color (state)” and tried both badges and entity cards, the colour stays blue regardless of whether the container is running or exited. However if I look at the history graph of that entity, exited shows up as purple.

Is this an issue with the integration or can I define the colour of a state in the config.yaml?