[search] Fixed failing retrieval's test.

1 files changed, 3 insertions, 3 deletions

diff --git a/search/integration_tests/retrieval_test.cpp b/search/integration_tests/retrieval_test.cpp
index 08511a1f7b..fc1361c128 100644
--- a/search/integration_tests/retrieval_test.cpp
+++ b/search/integration_tests/retrieval_test.cpp
@@ -219,14 +219,14 @@ UNIT_TEST(Retrieval_3Mwms)
   }
 
   {
-    TestCallback callback(mskHandle.GetId());
+    MultiMwmCallback callback({mskHandle.GetId(), mtvHandle.GetId(), zrhHandle.GetId()});
     search::Retrieval::Limits limits;
     limits.SetMaxNumFeatures(10 /* more than total number of features in all these mwms */);
 
     retrieval.Init(index, m2::RectD(m2::PointD(-1.0, -1.0), m2::PointD(1.0, 1.0)), params, limits);
     retrieval.Go(callback);
-    TEST(callback.WasTriggered(), ());
-    TEST_EQUAL(callback.Offsets().size(), 3 /* total number of features in all these mwms */, ());
+    TEST_EQUAL(3 /* total number of mwms */, callback.GetNumMwms(), ());
+    TEST_EQUAL(3 /* total number of features in all these mwms */, callback.GetNumFeatures(), ());
   }
 
   {