Total row should have same color according to the stage respectively

Summary:

Total row should have same color based on the stage. Kindly check the screenshot.

Content (please ensure you mask any confidential information):

Version (include the version you are using, if applicable):

Code Snippet (add any code snippets that support your topic, if applicable):