Fix HighlightAgent unit tests

Fix HighlightAgent unit tests in ResearchLiveHTML branch