Client command color highlight only highlights elements in the last command context

In a command rendered, all arguments before the last redirection are not rendered in color.

For example, the @s in

execute as @s

will render with highlighting color, but that in

execute as @s run help

won't.

Not sure if this is a bug or working as intended.