Trace Count Inconsistency, Combat vs Normal

Normal traces show the number of returns at the top of the screen during the "fight".

Combat traces only show the number of returns at the start, and then it goes away.

If I am low on spell energy, and near the end of an event, I want to focus only on the traces that I need to return; missing the current count makes this harder.