[Test] Update annotations test for new algorithm