Account color in unified inbox

In the previous version, you could see which account a message was in via a color code. The unified inbox now shows all messages the same. Am I doing something wrong, or was this feature removed?

The color chips are still displayed, but only if the message list contains messages from more than one account.

Thank you. It turns out that I wasn’t actually on the unified inbox. When I switched, they do appear. That said, the color chips are very small. Is there a reason they don’t span the full height of the message?
Thank you.