Color of battle timeline icons and castle sprites

Maybe this has been brought up or will be reworked in the future, but the colors of the battle timeline icons and castle sprites are not reflecting the palette of my fusing choice.
My chaos giant is red so I glance over the colors in the timeline to easily spot him but if hes not red he’s easy to miss.